Update app.py

app.py

@@ -30,19 +30,14 @@ disease_details = {
 # Passwords
 doctor_password = "doctor123"
 
-# Load doctor consultation model and tokenizer
-try:
-    doctor_tokenizer = AutoTokenizer.from_pretrained("ahmed-7124/dgptAW")
-except Exception as e:
-    print(f"Fast tokenizer failed: {e}. Falling back to slow tokenizer.")
-    doctor_tokenizer = AutoTokenizer.from_pretrained("ahmed-7124/dgptAW", use_fast=False)
-
-doctor_model = AutoModelForCausalLM.from_pretrained("ahmed-7124/dgptAW")
+# Load doctor consultation model and tokenizer (Now using the new model)
+tokenizer = AutoTokenizer.from_pretrained("harishussain12/PastelMed")
+model = AutoModelForCausalLM.from_pretrained("harishussain12/PastelMed")
 
 def consult_doctor(prompt):
-    inputs = doctor_tokenizer(prompt, return_tensors="pt")
-    outputs = doctor_model.generate(**inputs, max_new_tokens=100)
-    response = doctor_tokenizer.decode(outputs[0], skip_special_tokens=True)
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(**inputs, max_new_tokens=100)
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response
 
 # Functions