Spaces:

Unggi
/

hate_speech_kor

Runtime error

Unggi commited on Mar 18, 2023

Commit

d583822

1 Parent(s): 9868d11

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,8 +18,8 @@ def load_model(model_name):
 def inference(prompt):
-    model_name = "Unggi/hate_speech_bert"
     model, tokenizer = load_model(
         model_name = model_name
     )

 def inference(prompt):
+    model_name = "Unggi/ko_hate_speech_KcELECTRA" #"Unggi/hate_speech_bert"
     model, tokenizer = load_model(
         model_name = model_name
     )