bert-suicide-detection-hk-large

Running

wcyat commited on Jun 6

Commit

92836c7

•

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/bert-suicide-detection-hk-large")
 def classify_text(text):
     # Get predictions