Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, Text2TextGenerationPipeline
 
-pipe = pipeline(model = AutoModelForSeq2SeqLM.from_pretrained("jpelhaw/t5-word-sense-disambiguation"),tokenizer = AutoTokenizer.from_pretrained("jpelhaw/t5-word-sense-disambiguation"), )
+pipe = Text2TextGenerationPipeline(model = AutoModelForSeq2SeqLM.from_pretrained("jpelhaw/t5-word-sense-disambiguation"),tokenizer = AutoTokenizer.from_pretrained("jpelhaw/t5-word-sense-disambiguation"), )
 
 gr.Interface.from_pipeline(pipe,
             inputs = "text",