Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
|
@@ -43,8 +43,8 @@ import os
|
|
| 43 |
|
| 44 |
|
| 45 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 46 |
-
MODEL_DIR = "
|
| 47 |
-
TOKENIZER_DIR = "
|
| 48 |
# Load the model and tokenizer
|
| 49 |
try:
|
| 50 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_DIR)
|
|
|
|
| 43 |
|
| 44 |
|
| 45 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 46 |
+
MODEL_DIR = "./my_model"
|
| 47 |
+
TOKENIZER_DIR = "./my_tokenizer"
|
| 48 |
# Load the model and tokenizer
|
| 49 |
try:
|
| 50 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_DIR)
|