Spaces:

mskov
/

Speech-Trigger-Detection

Runtime error

mskov commited on Oct 10, 2023

Commit

3cdde10

1 Parent(s): 3df60e5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,10 +34,10 @@ emo_dict = {
 # static classes for now, but it would be best ot have the user select from multiple, and to enter their own
 class_options = {
-    "racism": ["racism", "hate speech", "bigotry", "racially targeted", "racial slur", "ethnic slur", "ethnic hate", "pro-white nationalism"],
-    "LGBTQ+ hate": ["gay slur", "trans slur", "homophobic slur", "transphobia", "anti-LBGTQ+"],
-    "sexually explicit": ["sexually explicit", "sexually coercive", "sexual exploitation", "vulgar", "raunchy", "sexist", "sexually demeaning", "sexual violence", "victim blaming"],
-    "alcohol use": ["alcohol", "drinking", "drinks", "under the influence", "liquor", "beer", "wine"]
 }
 pipe = pipeline("automatic-speech-recognition", model="openai/whisper-large")

 # static classes for now, but it would be best ot have the user select from multiple, and to enter their own
 class_options = {
+    "Racism": ["racism", "hate speech", "bigotry", "racially targeted", "racial slur", "ethnic slur", "ethnic hate", "pro-white nationalism"],
+    "LGBTQ+ Hate": ["gay slur", "trans slur", "homophobic slur", "transphobia", "anti-LBGTQ+"],
+    "Sexually Explicit": ["sexually explicit", "sexually coercive", "sexual exploitation", "vulgar", "raunchy", "sexist", "sexually demeaning", "sexual violence", "victim blaming"],
+    "Pregnancy & Birth Complications": ["miscarriage", "child loss", "child death", "abortion", "pregnancy", "childbirth", "baby shower", "postpartum"]
 }
 pipe = pipeline("automatic-speech-recognition", model="openai/whisper-large")