Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from huggingface_hub import HfApi, HfFolder
|
|
11 |
MODEL_NAME = "distilbert-base-uncased-finetuned-sst-2-english" # Change if needed
|
12 |
HF_DATASET_REPO = "M2ai/mgtd-logs" # Must be created beforehand
|
13 |
# Token from environment in Spaces
|
14 |
-
HF_TOKEN = os.getenv("
|
15 |
# Load model + tokenizer
|
16 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
17 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
|
|
|
11 |
MODEL_NAME = "distilbert-base-uncased-finetuned-sst-2-english" # Change if needed
|
12 |
HF_DATASET_REPO = "M2ai/mgtd-logs" # Must be created beforehand
|
13 |
# Token from environment in Spaces
|
14 |
+
HF_TOKEN = os.getenv("Mgtd")
|
15 |
# Load model + tokenizer
|
16 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
17 |
model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
|