Spaces:
Sleeping
Sleeping
Ashwin B
commited on
Commit
·
d5e81f8
1
Parent(s):
2d1282b
Fixing model path
Browse files- app/app.py +1 -1
app/app.py
CHANGED
@@ -20,7 +20,7 @@ GOEMOTIONS_LABELS = [
|
|
20 |
# -----------------------------
|
21 |
# Load model and tokenizer
|
22 |
# -----------------------------
|
23 |
-
MODEL_PATH = "
|
24 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH, device_map="auto", low_cpu_mem_usage=True)
|
25 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
|
26 |
print("Model device:", next(model.parameters()).device)
|
|
|
20 |
# -----------------------------
|
21 |
# Load model and tokenizer
|
22 |
# -----------------------------
|
23 |
+
MODEL_PATH = "outputs/model"
|
24 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH, device_map="auto", low_cpu_mem_usage=True)
|
25 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
|
26 |
print("Model device:", next(model.parameters()).device)
|