Update virtualhealth.py

virtualhealth.py

@@ -21,10 +21,12 @@ stop_words = set(stopwords.words("english"))
 # ============================
 # 🔹 1. Load Pretrained Medical Q&A Model
 # ============================
-qa_model_name = "deepset/roberta-base-squad2"  # Better model for medical Q&A
-tokenizer = AutoTokenizer.from_pretrained(qa_model_name)
-qa_model = AutoModelForQuestionAnswering.from_pretrained(qa_model_name)
-
+# qa_model_name = "deepset/roberta-base-squad2"  # Better model for medical Q&A
+# tokenizer = AutoTokenizer.from_pretrained(qa_model_name)
+# qa_model = AutoModelForQuestionAnswering.from_pretrained(qa_model_name)
+model_name = "dmis-lab/biobert-large-cased-v1.1-squad"  # ✅ Updated Model
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+qa_model = AutoModelForQuestionAnswering.from_pretrained(model_name)
 # ============================
 # 🔹 2. Load Symptom Checker Model
 # ============================
@@ -101,8 +103,10 @@ def get_medical_answer(question):
         tokenizer.convert_ids_to_tokens(inputs["input_ids"][0][answer_start:answer_end])
     )
 
-    return answer if answer.strip() and answer != "[CLS]" else "No reliable answer found."
-
+    if answer.strip() in ["", "[CLS]", "<s>"]:
+        return "I'm not sure. Please consult a medical professional."
+    
+    return answer
 # ============================
 # 🔹 9. Book a Doctor's Appointment
 # ============================