yasserrmd commited on
Commit
960ef8f
·
verified ·
1 Parent(s): 0842ac4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -40,8 +40,7 @@ class SinaReasonMedicalChat:
40
  try:
41
  print(f"Loading medical model: {MODEL_NAME}")
42
  self.tokenizer = AutoTokenizer.from_pretrained(
43
- MODEL_NAME,
44
- trust_remote_code=True
45
  )
46
 
47
  # Add padding token if not present
@@ -112,7 +111,7 @@ class SinaReasonMedicalChat:
112
  # Tokenize input
113
  inputs = self.tokenizer(
114
  text=prompt,
115
- images=None, # Required for this multimodal architecture
116
  return_tensors="pt"
117
  ).to(DEVICE)
118
 
 
40
  try:
41
  print(f"Loading medical model: {MODEL_NAME}")
42
  self.tokenizer = AutoTokenizer.from_pretrained(
43
+ MODEL_NAME
 
44
  )
45
 
46
  # Add padding token if not present
 
111
  # Tokenize input
112
  inputs = self.tokenizer(
113
  text=prompt,
114
+ #images=None, # Required for this multimodal architecture
115
  return_tensors="pt"
116
  ).to(DEVICE)
117