Gregoryjr commited on
Commit
d90a15e
1 Parent(s): e862b67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ if con:
30
 
31
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
32
  model = AutoModelForSequenceClassification.from_pretrained("Greys/milestonemodel")
33
- my_list = ['toxic', 'severe_toxic', 'obscene', 'threat', 'insult', 'identity_hate']
34
  def classify_sentence(text):
35
  inputs = tokenizer(text, return_tensors="pt")
36
  outputs = model(**inputs)
 
30
 
31
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
32
  model = AutoModelForSequenceClassification.from_pretrained("Greys/milestonemodel")
33
+ my_list = [' ','toxic', 'severe_toxic', 'obscene', 'threat', 'insult', 'identity_hate']
34
  def classify_sentence(text):
35
  inputs = tokenizer(text, return_tensors="pt")
36
  outputs = model(**inputs)