munnae commited on
Commit
324d648
·
verified ·
1 Parent(s): 4088cc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ st.set_page_config(page_title="Food Image Classifier", layout="centered")
11
  @st.cache_resource
12
  def load_model():
13
  st.text("Loading model...")
14
- model = pipeline("image-classification", model="munnae/bc220")
 
15
  st.text("Model loaded successfully!")
16
  return model
17
 
 
11
  @st.cache_resource
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