bert-suicide-detection-hk-new

Sleeping

wcyat commited on Jan 7

Commit

cd55d80

verified ·

1 Parent(s): 92836c7

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-large")
 def classify_text(text):
     # Get predictions

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