Spaces:

Destrobot
/

food-classifer

Sleeping

Destrobot commited on Feb 20

Commit

90afdc9

verified ·

1 Parent(s): 9f6857c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from transformers import pipeline
 model_id = "SABR22/food_models"
-pipe = pipeline("image-classification", model="SABR22/food_models")
 def classify_food(image):
     image = pipe(image)

 from transformers import pipeline
 model_id = "SABR22/food_models"
+pipe = pipeline("image-classification", model="Shresthadev403/food-image-classification")
 def classify_food(image):
     image = pipe(image)