risingodegua commited on
Commit
0713a95
1 Parent(s): 85574a0

Update app and requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +2 -0
app.py CHANGED
@@ -3,9 +3,9 @@ import numpy as np
3
 
4
  from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
5
 
6
- tokenizer = AutoTokenizer.from_pretrained("uhhlt/bert-based-uncased-hatespeech-movies")
7
 
8
- model = TFAutoModelForSequenceClassification.from_pretrained("uhhlt/bert-based-uncased-hatespeech-movies")
9
 
10
  def make_prediction(text):
11
  '''
3
 
4
  from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
5
 
6
+ tokenizer = AutoTokenizer.from_pretrained("risingodegua/hate-speech-detector")
7
 
8
+ model = TFAutoModelForSequenceClassification.from_pretrained("risingodegua/hate-speech-detector")
9
 
10
  def make_prediction(text):
11
  '''
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
1
+ huggingface-hub==0.0.18
2
+ huggingface-transformers==0.0.18