Dario Lopez Padial commited on
Commit
cb110ca
1 Parent(s): a3a02f1

Add application file flag

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-b
7
  hf_writer = gr.HuggingFaceDatasetSaver(os.environ['HF_API_TOKEN'], "text-classification-mistakes")
8
 
9
 
10
- def detect_suicide_comment(comentario="Quiero acabar con todo. No merece la pena vivir."):
11
- return pipe(comentario)
12
 
13
 
14
  iface = gr.Interface(
 
7
  hf_writer = gr.HuggingFaceDatasetSaver(os.environ['HF_API_TOKEN'], "text-classification-mistakes")
8
 
9
 
10
+ def detect_suicide_comment(input):
11
+ return pipe(input)
12
 
13
 
14
  iface = gr.Interface(