Update app.py

app.py

@@ -1,13 +1,16 @@
 import gradio as gr
+from transformers import pipeline
 
 
 ### https://huggingface.co/blog/fine-tune-vit -->> the blog post how to fine tune ViT
 
-examples = ["example-leaf.jpeg"]
 
-iface = gr.Interface("huggingface/nateraw/vit-base-beans",
-inputs="image",
-examples=examples,
-outputs=gr.outputs.Label(num_top_classes=5, label='Prediction'))
 
-iface.launch()
+pipe = pipeline(task="image-classification", 
+                model="nateraw/vit-base-beans")
+                
+gr.Interface.from_pipeline(pipe, 
+                           title="Image Classification",
+                           description="Test",
+                           examples=["example-leaf.jpeg"],
+                           ).launch()