|
import streamlit as st |
|
from PIL import Image |
|
|
|
from transformers import pipeline |
|
import numpy as np |
|
from transformers import AutoFeatureExtractor |
|
from transformers import AutoModelForImageClassification |
|
|
|
st.set_page_config(layout='wide', |
|
page_title='Food Category Classification & Recipes' |
|
) |
|
|
|
|
|
sidebar_acc = ['App Description', 'About Project'] |
|
sidebar_acc_nav = st.sidebar.radio('**INFORMATION SECTION**', sidebar_acc) |
|
|
|
if sidebar_acc_nav == 'App Description': |
|
st.sidebar.markdown("<h2 style='text-align: center;'> Food Category Classification Description </h2> ", unsafe_allow_html=True) |
|
st.sidebar.markdown("This is a Food Category Image Classifier model that has been trained by [Kaludi](https://huggingface.co/Kaludi) to recognize **12** different categories of foods, which includes **Bread**, **Dairy**, **Dessert**, **Egg**, **Fried Food**, **Fruit**, **Meat**, **Noodles**, **Rice**, **Seafood**, **Soup**, and **Vegetable**. It can accurately classify an image of food into one of these categories by analyzing its visual features. This model can be used by food bloggers, restaurants, and recipe websites to quickly categorize and sort their food images, making it easier to manage their content and provide a better user experience.") |
|
|
|
elif sidebar_acc_nav == 'About Project': |
|
st.sidebar.markdown("<h2 style='text-align: center;'> About Project </h2>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<hr style='text-align: center;'>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<h3 style='text-align: center;'>Project Location:</h3>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<p style='text-align: center;'><strong><a href='https://huggingface.co/Kaludi/food-category-classification-v2.0'>Model</a></strong> | <strong><a href='https://huggingface.co/datasets/Kaludi/food-category-classification-v2.0'>Dataset</a></strong></p>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<hr style='text-align: center;'>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<h3 style='text-align: center;'>Project Creators:</h3>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<p style='text-align: center;'><a href='https://github.com/Kaludii'><strong>AA</strong></a></p>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<p style='text-align: center;'><a href='https://github.com/Kaludii'><strong>AM</strong></a></p>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<p style='text-align: center;'><a href='https://github.com/Kaludii'><strong>BK</strong></a></p>", unsafe_allow_html=True) |
|
st.sidebar.markdown("<p style='text-align: center;'><a href='https://github.com/Kaludii'><strong>DK</strong></a></p>", unsafe_allow_html=True) |
|
|
|
|
|
def main(): |
|
st.title("Food Category Classification & Recipes") |
|
|
|
st.markdown("### Backgroud") |
|
st.markdown("This is a Food Category Image Classifier model that has been trained by [Kaludi](https://huggingface.co/Kaludi) to recognize **12** different categories of foods, which includes **Bread**, **Dairy**, **Dessert**, **Egg**, **Fried Food**, **Fruit**, **Meat**, **Noodles**, **Rice**, **Seafood**, **Soup**, and **Vegetable**. It can accurately classify an image of food into one of these categories by analyzing its visual features. This model can be used by food bloggers, restaurants, and recipe websites to quickly categorize and sort their food images, making it easier to manage their content and provide a better user experience.") |
|
st.header("Try it out!") |
|
|
|
images = ["examples/example_0.jpg", |
|
"examples/example_1.jpg", |
|
"examples/example_2.jpg", |
|
"examples/example_3.jpg", |
|
"examples/example_4.jpg", |
|
"examples/example_5.jpg", |
|
"examples/example_6.jpg", |
|
"examples/example_7.jpg"] |
|
show_images = False |
|
|
|
if st.checkbox("Show/Hide Examples"): |
|
|
|
|
|
show_images = not show_images |
|
if show_images: |
|
st.header("Example Images") |
|
for image in images: |
|
st.image(image, width=250) |
|
|
|
calories = st.slider("Select Max Calories (Not Functional Yet)", 50, 2000) |
|
|
|
|
|
st.text('Selected: {}'.format(calories)) |
|
|
|
uploaded_file = st.file_uploader("Upload Files",type=['png','jpeg','jpg']) |
|
|
|
if uploaded_file!=None: |
|
img=Image.open(uploaded_file) |
|
|
|
extractor = AutoFeatureExtractor.from_pretrained("Kaludi/food-category-classification-v2.0") |
|
model = AutoModelForImageClassification.from_pretrained("Kaludi/food-category-classification-v2.0") |
|
|
|
inputs = extractor(img,return_tensors="pt") |
|
outputs = model(**inputs) |
|
label_num=outputs.logits.softmax(1).argmax(1) |
|
label_num=label_num.item() |
|
|
|
probs = outputs.logits.softmax(dim=1) |
|
percentage = round(probs[0, label_num].item() * 100, 2) |
|
|
|
st.write("The Predicted Classification is:") |
|
|
|
if label_num==0: |
|
st.write("**Bread** (" + f"{percentage}%)") |
|
elif label_num==1: |
|
st.write("**Dairy** (" + f"{percentage}%)") |
|
elif label_num==2: |
|
st.write("Dessert (" + f"{percentage}%)") |
|
elif label_num==3: |
|
st.write("Egg (" + f"{percentage}%)") |
|
elif label_num==4: |
|
st.write("Fried Food (" + f"{percentage}%)") |
|
elif label_num==5: |
|
st.write("Fruit (" + f"{percentage}%)") |
|
elif label_num==6: |
|
st.write("Meat (" + f"{percentage}%)") |
|
elif label_num==7: |
|
st.write("Noodles (" + f"{percentage}%)") |
|
elif label_num==8: |
|
st.write("Rice (" + f"{percentage}%)") |
|
elif label_num==9: |
|
st.write("Seafood (" + f"{percentage}%)") |
|
elif label_num==10: |
|
st.write("Soup (" + f"{percentage}%)") |
|
else: |
|
st.write("Vegetable (" + f"{percentage}%)") |
|
|
|
|
|
select_health = st.selectbox("Select One (Not Functional Yet):", ["Choose Healthy or Non-Healthy", "Healthy", "Non-Healthy"]) |
|
if select_health == "Healthy": |
|
st.write("You selected healthy for", "**Bread**" if label_num==0 else "Dairy" if label_num==1 else "Dessert" if label_num==2 else "Egg" if label_num==3 else "Fried Food" if label_num==4 else "Fruit" if label_num==5 else "Meat" if label_num==6 else "Noodles" if label_num==7 else "Rice" if label_num==8 else "Seafood" if label_num==9 else "Soup" if label_num==10 else "Vegetable") |
|
|
|
elif select_health == "Non-Healthy": |
|
st.write("You selected non-healthy for", "**Bread**" if label_num==0 else "Dairy" if label_num==1 else "Dessert" if label_num==2 else "Egg" if label_num==3 else "Fried Food" if label_num==4 else "Fruit" if label_num==5 else "Meat" if label_num==6 else "Noodles" if label_num==7 else "Rice" if label_num==8 else "Seafood" if label_num==9 else "Soup" if label_num==10 else "Vegetable") |
|
|
|
|
|
st.image(img) |
|
|
|
if __name__ == '__main__': |
|
main() |
|
|
