Spaces:

Boboiazumi
/

IMPHNEN-AUTOMOD

Sleeping

BoboiAzumi commited on Sep 22

Commit

90e6910

1 Parent(s): 65bd12f

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ else:
 session = ort.InferenceSession("imphnenbert.onnx", providers=sess_providers)
 print("Loaded ONNX session")
-label = ["NORMAL", "PEDOFILIA", "SEXUAL_HARASSMENT"]
 def predict(text):
     encoded = tokenizer(text, padding=True, truncation=True, return_tensors="np")

 session = ort.InferenceSession("imphnenbert.onnx", providers=sess_providers)
 print("Loaded ONNX session")
+label = ["NORMAL", "PDF", "SH"]
 def predict(text):
     encoded = tokenizer(text, padding=True, truncation=True, return_tensors="np")