Spaces:

kompiangg
/

hate-speech-classification

Runtime error

kompiangg commited on Dec 26, 2023

Commit

2de5189

•

1 Parent(s): 0055aea

change model

Files changed (1) hide show

main.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformer import transformer
 import sys
-hate_speech_model = model.load_hugging_face_model('model_svm.pkl')
 hate_speech_dataset = dataset.load_dataset('data_clean.csv')
 tfidf = transformer.create_tfidf(hate_speech_dataset, 'Tweet', 'U')

 import sys
+hate_speech_model = model.load_hugging_face_model('model_rf.pkl')
 hate_speech_dataset = dataset.load_dataset('data_clean.csv')
 tfidf = transformer.create_tfidf(hate_speech_dataset, 'Tweet', 'U')