Update app.py

app.py

@@ -1,10 +1,14 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 
-pipe = pipeline(task="image-classification", 
-                 # model that can do 22k-category classification
-                 model="microsoft/beit-base-patch16-224-pt22k-ft22k")
+tokenizer = AutoTokenizer.from_pretrained("google/pegasus-xsum")
+
+model = AutoModelForSeq2SeqLM.from_pretrained("google/pegasus-xsum")
+
+pipe = pipeline(task="summarization", 
+                model=model,
+                tokenizer=tokenizer)
 gr.Interface.from_pipeline(pipe, 
-                           title="22k Image Classification",
-                           description="Object Recognition using Microsoft BEIT",
+                           title="Summarization using pegasus",
+                           description="Summarization description",
                            allow_flagging="never").launch(inbrowser=True)