Spaces:

SeyedAli
/

Food-Image-Classification

Runtime error

SeyedAli commited on Sep 21, 2023

Commit

c4a9309

•

1 Parent(s): a489376

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
 import gradio as gr
 from  PIL import Image
 import torch
-from transformers import ViTFeatureExtractor,pipeline
-model = ViTFeatureExtractor.from_pretrained('SeyedAli/Food-Image-Classification-VIT')
 def FoodClassification(image):
     pipline = pipeline(task="image-classification", model=model)

 import gradio as gr
 from  PIL import Image
 import torch
+from transformers import ViTImageProcessor,pipeline
+model = ViTImageProcessor.from_pretrained('SeyedAli/Food-Image-Classification-VIT')
 def FoodClassification(image):
     pipline = pipeline(task="image-classification", model=model)