Update app.py

app.py

@@ -36,12 +36,13 @@ def predict(question, context):
     input = f"Question: {question} Context: {context} Answer:"
     predictions = pipe(input)
     output = predictions[0]["generated_text"].replace("Question", "")
+    output = output.replace("Answer", "")
     return output
 
 demo = gr.Interface(
     predict,
     inputs=[gr.Textbox(lines=2, placeholder="Please provide your question", label="Question"), gr.Textbox(lines=2, placeholder="Please provide your context", label="Context")],
-    outputs=gr.Textbox(lines=2, placeholder="Predicted Answers..."),
+    outputs=gr.Textbox(lines=2, placeholder="Answers...", label="Output"),
     title="Question Answering",
 )