desorenia commited on
Commit
60a771d
1 Parent(s): 490e972

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, device=
20
  text = st.text_area("enter the text")
21
 
22
  if text:
23
- out = pipe(text)
24
  score = out['score'] * 4 - 2
25
  if score >= 0.5:
26
  label = 'not OK'
 
20
  text = st.text_area("enter the text")
21
 
22
  if text:
23
+ out = pipe(text)[0]
24
  score = out['score'] * 4 - 2
25
  if score >= 0.5:
26
  label = 'not OK'