munnae commited on
Commit
00b857d
·
verified ·
1 Parent(s): 0b730b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,14 +10,14 @@ st.set_page_config(page_title="Food Image Classifier", layout="centered")
10
  @st.cache_resource
11
  def load_model():
12
  st.text("Loading model...")
13
- model = pipeline("image-classification", model="google/mobilenet_v2_1.0_224")
14
  st.text("Model loaded successfully!")
15
  return model
16
 
17
  classifier = load_model()
18
 
19
  # Streamlit UI
20
- st.title("🍕🥖 Food Image Classifier")
21
  st.write("Upload an image of **roti, pizza, naan, or tofu** to classify.")
22
 
23
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
10
  @st.cache_resource
11
  def load_model():
12
  st.text("Loading model...")
13
+ model = pipeline("image-classification", model="munnae/bc220")
14
  st.text("Model loaded successfully!")
15
  return model
16
 
17
  classifier = load_model()
18
 
19
  # Streamlit UI
20
+ st.title("Virtual University FYP: Food Image Classifier")
21
  st.write("Upload an image of **roti, pizza, naan, or tofu** to classify.")
22
 
23
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])