Update classifier.py

updated function to include threshold

classifier.py

@@ -2,7 +2,7 @@ import spaces
 from transformers import pipeline
 
 #@spaces.GPU(duration=60)
-def classify(tweet, event_model, hftoken):
+def classify(tweet, event_model, hftoken, threshold):
     
     # event type prediction
     event_predictor = pipeline(task="text-classification", model=event_model, 
@@ -13,7 +13,12 @@ def classify(tweet, event_model, hftoken):
     prediction = event_predictor(tweet, **tokenizer_kwargs)[0]
     
     results["text"] = tweet
-    results["event"] = prediction["label"]
-    results["score"] = prediction["score"]
+    
+    if prediction["label"] != "none" and round(prediction["score"], 2) <= threshold:
+        results["event"] = "none"
+        results["score"] = prediction["score"]
+    else:
+        results["event"] = prediction["label"]
+        results["score"] = prediction["score"]
 
     return results