Update app.py

app.py

@@ -8,8 +8,8 @@ model = AutoModelForSeq2SeqLM.from_pretrained("jpelhaw/t5-word-sense-disambiguat
 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:20])
+print(example[0:40])
 
-answer = model.generate(input_ids=example['input_ids'], 
-                                attention_mask=example['attention_mask'], 
+answer = model.generate(input_ids=example[0:40], 
+                                attention_mask=example[9], 
                                 max_length=135)