Spaces:

SeyedAli
/

Food-Image-Classification

Runtime error

SeyedAli commited on Sep 21, 2023

Commit

0c0cd61

1 Parent(s): 5bf9ffd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ model = ViTFeatureExtractor.from_pretrained('SeyedAli/Food-Image-Classification-
 def FoodClassification(image):
     pipline = pipeline(task="image-classification", model=model)
-    output=pipline(image)
     return output
 iface = gr.Interface(fn=FoodClassification, inputs="image", outputs="text")

 def FoodClassification(image):
     pipline = pipeline(task="image-classification", model=model)
+    output=pipline(model(image, return_tensors='pt'))
     return output
 iface = gr.Interface(fn=FoodClassification, inputs="image", outputs="text")