Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 26 |
|
| 27 |
# ββ Model Initialization ββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 28 |
print(f"[INFO] Loading model: {MODEL_NAME} | device: {DEVICE}")
|
| 29 |
-
|
|
|
|
| 30 |
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(DEVICE)
|
| 31 |
model.eval()
|
| 32 |
print("[INFO] Model ready.")
|
|
|
|
| 26 |
|
| 27 |
# ββ Model Initialization ββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 28 |
print(f"[INFO] Loading model: {MODEL_NAME} | device: {DEVICE}")
|
| 29 |
+
# Force use_fast=False to avoid the sentencepiece backend error
|
| 30 |
+
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_fast=False)
|
| 31 |
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(DEVICE)
|
| 32 |
model.eval()
|
| 33 |
print("[INFO] Model ready.")
|