roberta-suicide-detection-hk

Sleeping

wcyat commited on May 29, 2024

Commit

f75714e

verified ·

1 Parent(s): 9bfb23d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 # Load pre-trained sentiment-analysis pipeline
-classifier = pipeline('text-classification', model="wcyat/bert-suicide-detection-hk")
 def classify_text(text):
     # Get predictions

 from transformers import pipeline
 # Load pre-trained sentiment-analysis pipeline
+classifier = pipeline('text-classification', model="wcyat/roberta-suicide-detection-hk")
 def classify_text(text):
     # Get predictions