munnae commited on
Commit
78c2195
·
verified ·
1 Parent(s): 8c78908

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ st.set_page_config(page_title="Food Image Classifier", layout="centered")
12
  def load_model():
13
  st.text("Loading model...")
14
  #model = pipeline("image-classification", model="munnae/bc220")
15
- model = pipeline("image-classification", model="nateraw/food101")
16
  st.text("Model loaded successfully!")
17
  return model
18
 
 
12
  def load_model():
13
  st.text("Loading model...")
14
  #model = pipeline("image-classification", model="munnae/bc220")
15
+ model = pipeline("image-classification", model="dwililiya/food101-model-classification")
16
  st.text("Model loaded successfully!")
17
  return model
18