Update app.py

app.py

@@ -97,7 +97,7 @@ def generate_answer(question, context, ground, do_pretrained):
     ).input_ids
 
     # Decode the context back
-    decoded_context = tokenizer.decode(input_ids, skip_special_tokens=True)[len(f"question: {question} context: "):]
+    decoded_context = tokenizer.decode(input_ids[0], skip_special_tokens=True)[len(f"question: {question} context: "):]
 
     # Generate the answer
     with torch.no_grad():