Update app.py

app.py

@@ -10,7 +10,7 @@ hf_saver = gr.HuggingFaceDatasetSaver(HF_TOKEN, "wfh-problematic")
 
 # model path in hugginface
 model_path = "yabramuvdi/distilbert-wfh"
-tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
+tokenizer = DistilBertTokenizer.from_pretrained(model_path)
 model = DistilBertForSequenceClassification.from_pretrained(model_path)
 
 # create a pipeline for predictions
@@ -23,8 +23,8 @@ title = "Work From Home Predictor"
 description = "Demo application that predicts the pressence of work from home in any given sequence of text."
 article = ""    # text at the end of the app
 examples = [
-    ["This is a work from home position", 0.998],
-    ["This position does not allow working from home", 0.998],
+    ["This is a work from home position", 0.9],
+    ["This position does not allow working from home", 0.5],
     ]
 
 #%%