ahmed-7124 commited on
Commit
e218653
·
verified ·
1 Parent(s): 40018c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -30,8 +30,15 @@ disease_details = {
30
  # Passwords
31
  doctor_password = "doctor123"
32
 
33
- # Load doctor consultation model and tokenizer (Now using the new model)
34
- tokenizer = AutoTokenizer.from_pretrained("harishussain12/PastelMed")
 
 
 
 
 
 
 
35
  model = AutoModelForCausalLM.from_pretrained("harishussain12/PastelMed")
36
 
37
  def consult_doctor(prompt):
 
30
  # Passwords
31
  doctor_password = "doctor123"
32
 
33
+ from transformers import AutoTokenizer, AutoModelForCausalLM
34
+
35
+ try:
36
+ # Force using the slow tokenizer
37
+ tokenizer = AutoTokenizer.from_pretrained("harishussain12/PastelMed", use_fast=False)
38
+ except Exception as e:
39
+ print(f"Tokenizer error: {e}")
40
+ tokenizer = AutoTokenizer.from_pretrained("harishussain12/PastelMed", use_fast=False)
41
+
42
  model = AutoModelForCausalLM.from_pretrained("harishussain12/PastelMed")
43
 
44
  def consult_doctor(prompt):