Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import BartTokenizer, BartForConditionalGeneration
+from transformers import pipeline, BartTokenizer, BartForConditionalGeneration
 
 # Initialize the image classification pipeline
 classifier = pipeline("image-classification", model="google/vit-base-patch16-224")