Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|