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)
|