Update app.py

app.py

@@ -10,8 +10,8 @@ hf_saver = gr.HuggingFaceDatasetSaver(HF_TOKEN, "wfh-problematic")
 
 # model path in hugginface
 model_path = "yabramuvdi/distilbert-wfh"
-tokenizer = DistilBertTokenizer.from_pretrained(model_path, use_auth_token=True)
-model = DistilBertForSequenceClassification.from_pretrained(model_path, use_auth_token=True)
+tokenizer = DistilBertTokenizer.from_pretrained(model_path, use_auth_token=HF_TOKEN)
+model = DistilBertForSequenceClassification.from_pretrained(model_path, use_auth_token=HF_TOKEN)
 
 # create a pipeline for predictions
 classifier = TextClassificationPipeline(model=model,