Update app.py

app.py

@@ -1,15 +1,12 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
-tokenizer = AutoTokenizer.from_pretrained("jpelhaw/t5-word-sense-disambiguation")
-
-model = AutoModelForSeq2SeqLM.from_pretrained("jpelhaw/t5-word-sense-disambiguation")
+pipe = pipeline(tokenizer = AutoTokenizer.from_pretrained("jpelhaw/t5-word-sense-disambiguation"), model = AutoModelForSeq2SeqLM.from_pretrained("jpelhaw/t5-word-sense-disambiguation")
 
 input = 'question: which description describes the word " java " best in the following context? descriptions:[  " A drink consisting of an infusion of ground coffee beans "," a platform-independent programming language ", or " an island in Indonesia to the south of Borneo " ] context: I like to drink " java " in the morning .'
 
-example = tokenizer.tokenize(input, add_special_tokens=True)
-print(example[0:40])
-
-answer = model.generate(input_ids=example[0:40], 
-                                attention_mask=example[9], 
-                                max_length=135)
+gr.Interface.from_pipeline(pipe,
+            inputs = "text",
+            outputs= gr.outputs.TextBox(),
+            title = "word sense disambiguation", 
+            allow_flagging="never").launch(inbrowser=True)