Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="skylord/swin-finetuned-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="skylord/swin-finetuned-food101")
|
| 16 |
st.text("Model loaded successfully!")
|
| 17 |
return model
|
| 18 |
|