Update app.py

app.py

@@ -42,7 +42,7 @@ import gradio as gr
 def predict(question, context):
     input = f"Question: {question} Context: {context} Answer:"
     predictions = pipe(input)
-    output = predictions["generated_text"].replace("Question", "")
+    output = predictions[0]["generated_text"].replace("Question", "")
     return output
 
 demo = gr.Interface(