Spaces:
Sleeping
Sleeping
Commit
·
bd96aca
1
Parent(s):
f3c8e25
added authentication
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ hf_saver = gr.HuggingFaceDatasetSaver(HF_TOKEN, "wfh-problematic")
|
|
10 |
|
11 |
# model path in hugginface
|
12 |
model_path = "yabramuvdi/distilbert-wfh"
|
13 |
-
tokenizer = DistilBertTokenizer.from_pretrained(model_path)
|
14 |
-
model = DistilBertForSequenceClassification.from_pretrained(model_path)
|
15 |
|
16 |
# create a pipeline for predictions
|
17 |
classifier = TextClassificationPipeline(model=model,
|
|
|
10 |
|
11 |
# model path in hugginface
|
12 |
model_path = "yabramuvdi/distilbert-wfh"
|
13 |
+
tokenizer = DistilBertTokenizer.from_pretrained(model_path, use_auth_token=True)
|
14 |
+
model = DistilBertForSequenceClassification.from_pretrained(model_path, use_auth_token=True)
|
15 |
|
16 |
# create a pipeline for predictions
|
17 |
classifier = TextClassificationPipeline(model=model,
|