Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,12 +25,15 @@ clf = joblib.load(clf_path)
|
|
| 25 |
|
| 26 |
# ========== Load DeBERTa Model + Tokenizer ==========
|
| 27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
deberta_model = AutoModelForSequenceClassification.from_pretrained(
|
| 29 |
"Sbhatti33/deberta_model"
|
| 30 |
).to(device)
|
| 31 |
|
| 32 |
-
deberta_tokenizer = AutoTokenizer.from_pretrained("Sbhatti33/deberta_model")
|
| 33 |
-
|
| 34 |
|
| 35 |
deberta_model.eval()
|
| 36 |
|
|
|
|
| 25 |
|
| 26 |
# ========== Load DeBERTa Model + Tokenizer ==========
|
| 27 |
|
| 28 |
+
# DeBERTa tokenizer and model from your repo
|
| 29 |
+
deberta_tokenizer = AutoTokenizer.from_pretrained(
|
| 30 |
+
"Sbhatti33/deberta_model",
|
| 31 |
+
use_fast=False # critical for crash
|
| 32 |
+
)
|
| 33 |
deberta_model = AutoModelForSequenceClassification.from_pretrained(
|
| 34 |
"Sbhatti33/deberta_model"
|
| 35 |
).to(device)
|
| 36 |
|
|
|
|
|
|
|
| 37 |
|
| 38 |
deberta_model.eval()
|
| 39 |
|