Update app.py

app.py

@@ -1,3 +1,10 @@
+"""
+Diabetes Version
+@aim: Demo for testing purposes only
+@inquiries: Dr M As'ad 
+@email: [email protected]
+"""
+
 # Import necessary libraries
 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, pipeline
@@ -41,9 +48,13 @@ st.sidebar.write(f"You're now chatting with **{selected_model}**")
 st.sidebar.image("https://www.hmgaihub.com/untitled.png")
 st.sidebar.markdown("*Generated content may be inaccurate or false.*")
 st.sidebar.markdown("*This is an under development project.*")
+st.sidebar.markdown("*Not a replacement for medical advice from a doctor.*")
+
+# Loading message placeholder
+loading_message = st.empty()
 
 def load_model(selected_model_name):
-    st.info("Loading the model, please wait...")
+    loading_message.info("Loading the model, please wait...")
     model_name = model_links[selected_model_name]
 
     # Define the BitsAndBytesConfig for quantization
@@ -81,13 +92,12 @@ def load_model(selected_model_name):
         "mistralai/Mistral-7B-Instruct-v0.2", trust_remote_code=True
     )
 
-    st.success("Model is ready. Now we are ready!")
+    # Clear the loading message
+    loading_message.success("Model is ready. Now we are ready!")
 
     return model, tokenizer
 
 
-
-
 # Load model and tokenizer
 model, tokenizer = load_model(selected_model)
 
@@ -106,11 +116,10 @@ if prompt := st.chat_input("Ask me anything about diabetes"):
     st.session_state.messages.append({"role": "user", "content": prompt})
 
     instructions = """
-    Act as a highly knowledgeable endocrinology doctor with expertise in explaining complex medical information in an understandable way to patients who do not have a medical background. Your responses should not only convey empathy and care but also demonstrate a high level of medical accuracy and reliability.
-you will answer only what the need and in professional way. do not add extra unnecessary information. you can however chat with the patient casually
+    Act as a highly knowledgeable doctor with special interest in diabetes, skilled at explaining complex medical information in a way that is easy to understand for patients without a medical background. Your responses should not only demonstrate empathy and care but also uphold a high standard of medical accuracy and reliability. Respond precisely to what the patient needs in a professional, accurate, and reassuring manner, avoiding any unnecessary information.
     """
 
-    full_prompt = f"<s>[INST] {instructions} [/INST] {prompt}</s>"
+    full_prompt = f"<s>[INST] {prompt} [/INST] {instructions}</s>"
 
     with st.chat_message("assistant"):
         result = pipeline(