Update src/streamlit_app.py

src/streamlit_app.py

@@ -42,10 +42,11 @@ generator = load_model()
 # -----------------------------
 # Streamlit UI
 # -----------------------------
-st.set_page_config(page_title="BioGPT — Biology QA Demo", layout="centered")
-st.title("🧬 BioGPT — Pubmed QA Demo")
+st.set_page_config(page_title="BioGPT — Pubmed Demo", layout="centered")
+st.title("🧬 BioGPT — Pubmed chatbot")
 
-st.write("Ask a biology-related question and get an answer generated by BioGPT-Large-PubMedQA fine-tuned with your Biology-QA adapter.")
+st.write("Ask a biology-related question and get an answer.")
+st.write("Generated by BioGPT-Large-PubMedQA fine-tuned on cais/mmlu and allenai/sciq datasets.")
 
 user_input = st.text_area("Enter your biology question:", height=150)
 
@@ -69,4 +70,4 @@ if st.button("Get Answer"):
         st.warning("Please enter a question.")
 
 st.markdown("---")
-st.caption("Model: microsoft/BioGPT-Large-PubMedQA + adapter kirubel1738/biogpt-biology-qa | Runs on CPU/GPU automatically")
+st.caption("Model: microsoft/BioGPT-Large-PubMedQA + adapter kirubel1738/biogpt-pubmedqa-finetuned | Runs on CPU automatically")