Spaces:
Runtime error
Runtime error
Commit
·
3febeb0
1
Parent(s):
3622bd1
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-b
|
|
| 8 |
|
| 9 |
|
| 10 |
def detect_suicide_comment(input):
|
| 11 |
-
return pipe(input)
|
| 12 |
|
| 13 |
|
| 14 |
iface = gr.Interface(
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
def detect_suicide_comment(input):
|
| 11 |
+
return pipe(input)[0]
|
| 12 |
|
| 13 |
|
| 14 |
iface = gr.Interface(
|