Update app.py

app.py

@@ -9,7 +9,7 @@ import transformers
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 def load_model(model_name):
-    # model_name = "Unggi/hate_speech_bert"
+    # model_name = "Unggi/hate_speech_classifier_KcElectra"
     # model
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     # tokenizer..
@@ -19,7 +19,7 @@ def load_model(model_name):
 
 
 def inference(prompt):
-    model_name = "Unggi/ko_hate_speech_KcELECTRA" #"Unggi/hate_speech_bert"
+    model_name = "Unggi/hate_speech_classifier_KcElectra" #"Unggi/hate_speech_bert"
     
     model, tokenizer = load_model(
         model_name = model_name