made the tokenizer and model local

app.py

@@ -42,7 +42,9 @@ model.eval()
 
 # === Role Agent with instruction/input/output format ===
 class RoleAgent:
-    def __init__(self, role_instruction):
+    def __init__(self, role_instruction, tokenizer, model):
+        self.tokenizer = tokenizer
+        self.model = model
         self.role_instruction = role_instruction
 
     def act(self, input_text):
@@ -102,16 +104,16 @@ def simulate_interaction(user_input, iterations=1):
     for i in range(iterations):
         # Summarize
         sum_in = "\n".join(history) + f"\nPrevious Vignette: {summary}"
-        sum_out = summarizer.act(sum_in)
+        sum_out = summarizer.act(sum_in, tokenizer, model)
         summary = sum_out["output"]
 
         # Diagnose
-        diag_out = diagnoser.act(summary)
+        diag_out = diagnoser.act(summary, tokenizer, model)
         diagnosis = diag_out["output"]
 
         # Question
         q_in = f"Vignette: {summary}\nCurrent Estimated Diagnosis: {diag_out['thinking']} {diagnosis}"
-        q_out = questioner.act(q_in)
+        q_out = questioner.act(q_in, tokenizer, model)
         history.append(f"Doctor: {q_out['output']}")
         history.append("Patient: (awaiting response)")