Update app.py

app.py

@@ -30,15 +30,14 @@ disease_details = {
 # Passwords
 doctor_password = "doctor123"
 
-# Load doctor consultation model and tokenizer
 # Load doctor consultation model and tokenizer
 try:
-    doctor_tokenizer = AutoTokenizer.from_pretrained("ahmed-7124/PastelMedAW")
+    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/PastelMedAW", use_fast=False)
+    doctor_tokenizer = AutoTokenizer.from_pretrained("ahmed-7124/dgptAW", use_fast=False)
 
-doctor_model = AutoModelForCausalLM.from_pretrained("ahmed-7124/PastelMedAW")
+doctor_model = AutoModelForCausalLM.from_pretrained("ahmed-7124/dgptAW")
 
 def consult_doctor(prompt):
     inputs = doctor_tokenizer(prompt, return_tensors="pt")
@@ -46,7 +45,6 @@ def consult_doctor(prompt):
     response = doctor_tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response
 
-
 # Functions
 def register_patient(name, age, gender, password):
     patient_id = len(patients_db) + 1
@@ -133,7 +131,7 @@ def doctor_dashboard(password):
 # Gradio Interfaces
 registration_interface = gr.Interface(
     fn=register_patient,
-    inputs=[
+    inputs=[ 
         gr.Textbox(label="Patient Name"),
         gr.Number(label="Age"),
         gr.Radio(label="Gender", choices=["Male", "Female", "Other"]),
@@ -150,7 +148,7 @@ pdf_extraction_interface = gr.Interface(
 
 report_analysis_interface = gr.Interface(
     fn=analyze_report,
-    inputs=[
+    inputs=[ 
         gr.Number(label="Patient ID"),
         gr.Textbox(label="Report Text"),
     ],
@@ -177,7 +175,7 @@ pharmacist_space_interface = gr.Interface(
 
 patient_dashboard_interface = gr.Interface(
     fn=patient_dashboard,
-    inputs=[
+    inputs=[ 
         gr.Number(label="Patient ID"),
         gr.Textbox(label="Password", type="password"),
     ],
@@ -227,4 +225,5 @@ with gr.Blocks() as app:
     with gr.Tab("Doctor Consult"):
         consult_doctor_interface.render()
 
+# Launch the app
 app.launch(share=True)