Update app.py

app.py

@@ -1,12 +1,11 @@
 import gradio as gr
+from transformers import pipeline
 
-model = gr.load("models/manoh2f2/finetuned-vit-flowers")
+pipe = pipeline("image-classification", model="manoh2f2/finetuned-vit-flowers")
 images = ["images/daisy.jpg", "images/dandelion.jpg", "images/rosa.jpg", "images/sunflower.jpg", "images/tulip.jpg"]
 
-iface = gr.Interface(
-        model, 
-        inputs = gr.Image(label="Upload Image"),
-        outputs = gr.Textbox(),
+iface = gr.Interface.from_pipeline(
+        pipe, 
         examples= [ [example] for example in images],
         description="Daisy, Dandelion, Rose, Sonflower, Tulip",
         title="FLower Classifier - Vit"