Update app.py

app.py

@@ -13,14 +13,17 @@ pipe_biogpt = pipeline("text-generation", model="microsoft/biogpt-large-pubmedqa
 
 title = "BioGPT Q&A Demo"
 description = """
+NOTE: this model requires the inputs in the following format: `question: <text> context: <text> answer: <text> target: the answer to the question given the context is`.
+
 Check out the [BioGPT-Large-PubMedQA model card](https://huggingface.co/microsoft/biogpt-large-pubmedqa) for more info. 
+
 **Disclaimer:** this demo was made for research purposes only and should not be used for medical purposes.
 """
 
 def inference(text):
   output_biogpt = pipe_biogpt(text, max_length=100)[0]["generated_text"]
   return [
-      output_biogpt[-1], 
+      output_biogpt.split(' ')[-1], 
   ]
 
 io = gr.Interface(