Update human_text_detect.py

human_text_detect.py

@@ -79,7 +79,7 @@ def get_threshold_obj(model_name, topic):
 
     return threshold
 
-def detect_human_text(model_name, topic, text):
+def detect_human_text(model_name, topic, threshold, text):
 
     torch.cuda.empty_cache()
     gc.collect()
@@ -157,7 +157,7 @@ def detect_human_text(model_name, topic, text):
 
         results = res['HC']
     
-    threshold = get_threshold_obj(model_name, topic)
+    # threshold = get_threshold_obj(model_name, topic)
     print(f"threshold: {threshold}, results: {results}")
     return round((results / threshold) - 1, 2), round(results, 2), res['sentences']