ytkoa commited on
Commit
3b379dc
·
verified ·
1 Parent(s): 458cac7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ threshold = config['predct']['threshold']
7
 
8
  def predict(text: str):
9
  prediction = model(text).item()
10
- label = "Negative" if prediction >= threshold else "Positive"
11
  return label, float(prediction)
12
 
13
  examples = [
 
7
 
8
  def predict(text: str):
9
  prediction = model(text).item()
10
+ label = "Токсичный❌" if prediction >= threshold else "Не токсичный✅"
11
  return label, float(prediction)
12
 
13
  examples = [