minemaster01 commited on
Commit
a37c9c1
·
verified ·
1 Parent(s): d4f370e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("HUGGINGFACE_TOKEN")
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)