Update app.py

app.py

@@ -1,14 +1,11 @@
 import gradio as gr
-from transformers import pipeline
 
-pipeline = pipeline(task="image-classification", model="google/vit-base-patch16-224")
+examples = [
+    'tiget.jpg',
+    'teapot.jpg',
+    'palace.jpg'
+]
 
-def predict(image):
-  predictions = pipeline(image)
-  return {p["label"]: p["score"] for p in predictions}
-
-gr.Interface(
-    predict,
-    inputs=gr.inputs.Image(type="filepath"),
-    outputs=gr.outputs.Label(num_top_classes=1),
-).launch()
+gr.Interface.load(
+    "google/vit-base-patch16-224",
+    examples=examples).launch()