Update app.py

app.py

@@ -1,13 +1,11 @@
+import transformers
 import gradio as gr
-from transformers import pipeline
 
-pipe = pipeline("image-classification", model="afern24/vit-beans-finetuned")
+model = transformers.VisionModel.from_pretrained("afern24/vit-beans-finetuned")
 
-def classify_image(filepath):
-    pred = pipe(filepath)
-    outputs = {}
-    for p in preds:
-        outputs[p["label"]] = p["score"]
-    return outputs
+def classify_image(image):
+  inputs = transformers.Image.from_path(image)
+  outputs = model(inputs)
+  return outputs
 
-gr.Interface(fn=classify_image, inputs=gr.inputs.Image(type="filepath"), outputs=gr.outputs.Label(num_top_classes=3), examples=[["leaf1.jpg"]]).load("models/afern24/vit-beans-finetuned").launch()
+gr.Interface(fn=classify_image, inputs=gr.inputs.Image, outputs=gr.outputs.Label(num_top_classes=3), examples=[["leaf1.jpg"]]).load("models/afern24/vit-beans-finetuned").launch()