vietnamese_hate_speech_detection

Sleeping

jesse-tong commited on Apr 12, 2025

Commit

6755cf9

1 Parent(s): 0ec27d9

Reduce threshold

Files changed (1) hide show

api.py CHANGED Viewed

@@ -40,7 +40,7 @@ def load_model_lstm():
     model = model.to(device)
     return model, device
-def inference(model, device, comments: str | list, threshold: float = 0.6):
     if isinstance(comments, str):
         comments = [comments]
     elif not isinstance(comments, list):

     model = model.to(device)
     return model, device
+def inference(model, device, comments: str | list, threshold: float = 0.5):
     if isinstance(comments, str):
         comments = [comments]
     elif not isinstance(comments, list):