Update app.py

app.py

@@ -1,7 +1,6 @@
 from transformers import pipeline
+import gradio as gr
 
-#pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
-pipe = pipeline(model="Salesforce/blip-image-captioning-base")
-preds = pipe(
-    images="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg"
-)
+pipe = pipeline("image-classification", model="google/vit-base-patch16-224")
+
+gr.Interface.from_pipeline(pipe).launch()