Update README.md

README.md

@@ -39,52 +39,81 @@ import torch
 # Clear CUDA memory before using the model
 torch.cuda.empty_cache()
 
-model_name = "syubraj/Phi-3.5-mini-instruct-MedicalChat"
-generator = pipeline("text-generation", model=model_name, device=-1)  # Use CPU
+model_name = "syubraj/MedicalChat-Phi-3.5-mini-instruct"
 
-def generate_response(user_query, system_message=None, max_length=256):
+try:
+    generator = pipeline("text-generation", model=model_name)
+    print("Pipeline loaded successfully.")
+except Exception as e:
+    raise ValueError(f"Error loading pipeline: {e}")
+
+def generate_response(user_query: str, system_message: str = None, max_length: int = 1024) -> str:
     """
     Generates a response based on the given user query.
-    
+
     :param user_query: The user's input message.
     :param system_message: Custom system instruction (optional, defaults to medical assistant).
     :param max_length: Max tokens to generate.
     :return: Generated assistant response.
     """
-    if system_message is None:
-        system_message = (
-            "You are a trusted AI-powered medical assistant. "
-            "Analyze patient queries carefully and provide accurate, professional, and empathetic responses. "
-            "Prioritize patient safety, adhere to medical best practices, and recommend consulting a healthcare provider when necessary."
-        )
+    if not user_query.strip():
+        return "Error: User query cannot be empty."
 
-    prompt = f"""<|system|> {system_message} <|end|> 
-                 <|user|> {user_query} <|end|> 
-                 <|assistant|>"""
+    if system_message is None:
+        system_message = ("You are a trusted AI-powered medical assistant. "
+                          "Analyze patient queries carefully and provide accurate, professional, and empathetic responses. "
+                          "Prioritize patient safety, adhere to medical best practices, and recommend consulting a healthcare provider when necessary.")
 
-    output = generator(prompt, max_length=max_length, truncation=True)
-    response = output[0]["generated_text"].split("<|assistant|>")[-1].strip().split("<|end|>")[0].strip()
+    prompt = f"<|system|> {system_message} <|end|> <|user|> {user_query} <|end|> <|assistant|>"
 
-    return response
+    try:
+        outputs = generator(prompt, max_length=max_length, truncation=True)
+        response = outputs[0]["generated_text"].split("<|assistant|>")[-1].strip().split("<|end|>")[0].strip()
+    except Exception as e:
+        return f"Error generating response: {e}"
+    
+    return response if response else "Error: No response generated."
 
-# Example Usage
 user_input = "Hi Doctor, How to increase my height? I am 23 years old."
-assistant_reply = generate_response(user_input)
-print("\nAssistant Response:\n", assistant_reply)
+response = generate_response(user_input)
+print("Assistant Response:", response)
+
 ```
 
 ### 📝 Example Output
-```text
-Assistant Response:
-Hello! I understand that you're interested in increasing your height, and it's great that you're seeking advice at a young age.
-Here are some points to consider:
 
-1. **Genetics**: The most significant factor in determining your height is genetics. You can't change your genetic makeup, but understanding your family's height patterns can give you an idea of your potential.
+Assistant Response: Hello! I understand that you're looking to potentially increase your height, and I'll do my best to provide you with helpful information.
 
-2. **Nutrition**: Ensure you're getting a balanced diet rich in calcium, protein, and vitamins D and A, which are essential for bone growth and development.
+Firstly, it's important to clarify that after puberty, the growth plates in our bones, known as epiphyseal plates, have typically fused, meaning our height generally stabilizes. For most adults, especially those who have reached their late teens or early twenties, further height increase is unlikely.
+
+However, there are a few factors that can influence your height:
+
+1. **Genetics**: Your genes play a significant role in determining your height. If your parents are tall, you might be taller too.
+
+2. **Nutrition**: Proper nutrition is essential for growth. Ensure you're getting a balanced diet with enough protein, calcium, and vitamins.
+
+3. **Posture**: Sometimes, poor posture can make you appear shorter. Practicing good posture might help you look taller.
+
+4. **Sleep**: Adequate sleep is crucial for growth hormones to work effectively.
+
+5. **Exercise**: Regular exercise can help maintain a healthy posture and spine alignment, which might help you appear taller.
+
+Here are some tips that might help you look taller:
+
+- **Stand Straight**: Keep your shoulders back and your back straight. This can make you appear taller.
+
+- **Practice Good Posture**: Be mindful of your posture throughout the day.
+
+- **Exercise**: Engage in exercises that strengthen your core, back, and leg muscles.
+
+- **Shoe Lifts**: Some people find that wearing shoe lifts can make them appear taller. However, this is more of a visual effect and won't actually increase your height.
+
+- **Physical Therapy**: If you have concerns about your posture or spinal alignment, a physical therapist can provide exercises and treatments to help.
+
+Remember, while these tips might help you look taller, they won't change your actual height. If you're concerned about your height for medical reasons, such as a growth hormone deficiency, it's best to consult with a healthcare provider.
+
+I hope this information is helpful! If you have any more questions or need further assistance, feel free to ask.
 
-3. **Healthy Habits**: Regular exercise, especially weight-bearing activities like walking, running, or resistance training, can help maintain bone health and potentially contribute.
-```
 
 ---