Update app.py

app.py

@@ -1,19 +1,5 @@
-import transformers
-from transformers import ViTModel, ViTImageProcessor
-from PIL import Image
 import gradio as gr
 
-
-model = ViTModel.from_pretrained("afern24/vit-beans-finetuned")
-processor = ViTImageProcessor.from_pretrained('google/vit-base-patch16-224-in21k')
-
-def classify_image(image):
-  inputs = processor(images=image, return_tensors="pt")
-  outputs = model(**inputs)
-  return outputs
-
-# Define your examples as a list of lists
-examples = [["leaf1.jpg"], ["leaf2.jpg"], ["leaf3.jpg"]]
-
-# Load your interface with examples
-gr.Interface(fn=classify_image, inputs=gr.Image(type="pil"), outputs=gr.outputs.Label, examples=examples).launch(debug=True)
+gr.Interface.load(
+             "afern24/vit-beans-finetuned",
+             examples=["leaf1.jpg", "leaf2.jpg", "leaf3.jpg"]).launch()