Update app.py

app.py

@@ -1,5 +1,5 @@
 from transformers import pipeline
 import gradio as gr
 
-pipeline = pipeline("text-generation", model="openai-community/gpt2")
+pipeline = pipeline("image-classification", model="google/vit-base-patch16-224")
 gr.Interface.from_pipeline(pipeline).launch()