Update app.py

app.py

@@ -1,17 +1,13 @@
 import gradio as gr
 import torch
-from transformers import AutoImageProcessor, AutoModelForImageClassification
+from transformers import pipeline
 
 model_id = "SABR22/food_models"
-processor = AutoImageProcessor.from_pretrained(model_id)
-model = AutoModelForImageClassification.from_pretrained(model_id)
+pipe = pipeline("image-classification", model="SABR22/food_models")
 
 def classify_food(image):
-    inputs = processor(images=image, return_tensors="pt")
-    outputs = model(**inputs)
-    logits = outputs.logits
-    predicted_class_idx = logits.argmax(-1).item()
-    return model.config.id2label[predicted_class_idx]
+    image = pipe(image)
+    return image[0]['label']
 
 interface = gr.Interface(
     fn=classify_food,