Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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'
|