MentalBERT output

app.py

@@ -6,7 +6,8 @@ vs_examples = [
 	["In child protective services, further providing for definitions, for immunity from liability"], 
 	["Lol what? Measles is a real thing. Get vaccinated"]]
 vs_title = "Vaccine Sentiment Task"
-vs_desc = "Enter text to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The three provided examples have true labels 'vaccine critical', 'neutral', 'vaccine supportive' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the VS2 dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+# vs_desc = "Enter text to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The three provided examples have true labels 'vaccine critical', 'neutral', 'vaccine supportive' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the VS2 dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+vs_desc = "Enter text to generate labels from 2 models (BERT, PHS-BERT). The three provided examples have true labels 'vaccine critical', 'neutral', 'vaccine supportive' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the VS2 dataset description in our paper (link provided in the corresponding Hugging Face repository)."
 
 hm_examples = [
 	["Serious as a heart attack question/thought Riddle me this. Why, oh why, does cold brew coffee get warm after sitting to long. Taste terrible. And Hot coffee get literally COLD after sitting too long. Tastes terrible. Like what. Why don't cold stay cold and hot only get warm?"], 
@@ -14,7 +15,8 @@ hm_examples = [
 	["The older we get the less likely we are to view a sudden massive heart attack as a tragedy than a stroke of good luck."],
 	["My son is not feeling well today."]]
 hm_title = "Health Mention Task"
-hm_desc = "Enter text to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The three provided examples have true labels 'Figurative/Hyperbolic Health Mentions', 'Non-personal Health Mentions', 'Personal Health Mentions' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the RHMD dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+# hm_desc = "Enter text to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The three provided examples have true labels 'Figurative/Hyperbolic Health Mentions', 'Non-personal Health Mentions', 'Personal Health Mentions' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the RHMD dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+hm_desc = "Enter text to generate labels from 2 models (BERT, PHS-BERT). The three provided examples have true labels 'Figurative/Hyperbolic Health Mentions', 'Non-personal Health Mentions', 'Personal Health Mentions' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the RHMD dataset description in our paper (link provided in the corresponding Hugging Face repository)."
 
 dep_examples = [
 	["Nothing major, some underage drinking, a little pot, but I wasn't a bad kid. Furthermore, I feel like I matured well and have gone on to be successful for my age (24) IMO. I've also had positive experiences with police. The resource officer at my high school was a really good guy and almost like a mentor to me. When I was 17 I was arrested when police raided a friends house because they were tipped off that there was to be some under-aged drinking. Furthermore, I told him before we got really serious that I have anxiety and depression, and if he can’t handle that then I’ll understand if we end things. But *he* chose to stay with me and help me if I wanted it. Honestly I’m just tired of defending myself to him after this fight. I shouldn’t have to. And he kept threatening to break up; then why don’t you do it?"],
@@ -30,7 +32,8 @@ covid_examples = [
 	["The latest Boarding Pass!  #budget2020 #coronavirus"]
 ]
 covid_title = "COVID Category Task"
-covid_desc = "Enter covid-related tweets to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The two provided examples have true labels 'news', 'personal narrative' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the Covid category dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+# covid_desc = "Enter covid-related tweets to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The two provided examples have true labels 'news', 'personal narrative' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the Covid category dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+covid_desc = "Enter covid-related tweets to generate labels from 2 models (BERT, PHS-BERT). The two provided examples have true labels 'news', 'personal narrative' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the Covid category dataset description in our paper (link provided in the corresponding Hugging Face repository)."
 
 # 2nd example looks like the most risk
 suicide_examples = [
@@ -47,14 +50,16 @@ stress_examples = [
 	["My daughter's father I was with for 5 years on and off. He was not abusive the first year, however when he became so I left and found out two weeks later I was pregnant. Of course he begged and pleaded for a second chance, and I believed that my daughter deserved me to at least give it a shot. I came back and surprise surprise it was worse than it ever was. I stuck it out until he disappeared for the eleventeenth time on a drunken bender, I checked my Facebook to find some scumbag girl who was dating one of his best friends was posting horrific, nasty, way out shit about me."]
 ]
 stress_title = "Stress Detection Task"
-stress_desc = "Enter text to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The two provided examples have true labels 'no stress', 'stress' respectively (NOTE: you may have to click submit again to generate the correct label)."# For more details, please refer to the Dreaddit dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+# stress_desc = "Enter text to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The two provided examples have true labels 'no stress', 'stress' respectively (NOTE: you may have to click submit again to generate the correct label)."# For more details, please refer to the Dreaddit dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+stress_desc = "Enter text to generate labels from 2 models (BERT, PHS-BERT). The two provided examples have true labels 'no stress', 'stress' respectively (NOTE: you may have to click submit again to generate the correct label)."# For more details, please refer to the Dreaddit dataset description in our paper (link provided in the corresponding Hugging Face repository)."
 
 other_examples = [
 	["@anxietyfighter suffered social anxiety for 4 yrs when i had my first panic attack,got worse when i went to uni, so have just started paxil"],
 	["@JessBarrett227 Taking someone off 150mg off Seroquel, mixing it with Olanzapine in 2 wks causes psychosis - a proper assess was not done. "]
 ]
 other_title = "Adverse Drug Reaction Task"
-other_desc = "Enter text to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The two provided examples have true labels 'no adverse drug reaction', 'adverse drug reaction' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the SMM4H T1 dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+# other_desc = "Enter text to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). The two provided examples have true labels 'no adverse drug reaction', 'adverse drug reaction' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the SMM4H T1 dataset description in our paper (link provided in the corresponding Hugging Face repository)."
+other_desc = "Enter text to generate labels from 2 models (BERT, PHS-BERT). The two provided examples have true labels 'no adverse drug reaction', 'adverse drug reaction' respectively (NOTE: you may have to click submit again to generate the correct labels)."# For more details, please refer to the SMM4H T1 dataset description in our paper (link provided in the corresponding Hugging Face repository)."
 
 # def make_interfaces(folder):
 # 	predictor_bert = ktrain.load_predictor(folder + "bert")
@@ -104,7 +109,12 @@ def vs_PHSBERT(text):
 vs_bert_io = Interface(fn=vs_BERT, inputs="text", outputs="text")
 vs_mental_io = Interface(fn=vs_MentalBERT, inputs="text", outputs="text")
 vs_phs_io = Interface(fn=vs_PHSBERT, inputs="text", outputs="text")
-vs = Parallel(vs_bert_io, vs_mental_io, vs_phs_io, 
+# vs = Parallel(vs_bert_io, vs_mental_io, vs_phs_io, 
+# 	examples=vs_examples, 
+# 	title=vs_title, 
+# 	description=vs_desc,
+# 	theme="peach")
+vs = Parallel(vs_bert_io, vs_phs_io, 
 	examples=vs_examples, 
 	title=vs_title, 
 	description=vs_desc,
@@ -138,7 +148,12 @@ def hm_PHSBERT(text):
 hm_bert_io = Interface(fn=hm_BERT, inputs="text", outputs="text")
 hm_mental_io = Interface(fn=hm_MentalBERT, inputs="text", outputs="text")
 hm_phs_io = Interface(fn=hm_PHSBERT, inputs="text", outputs="text")
-hm = Parallel(hm_bert_io, hm_mental_io, hm_phs_io, 
+# hm = Parallel(hm_bert_io, hm_mental_io, hm_phs_io, 
+# 	examples=hm_examples, 
+# 	title=hm_title, 
+# 	description=hm_desc,
+# 	theme="peach")
+hm = Parallel(hm_bert_io, hm_phs_io, 
 	examples=hm_examples, 
 	title=hm_title, 
 	description=hm_desc,
@@ -168,7 +183,12 @@ def covid_PHSBERT(text):
 covid_bert_io = Interface(fn=covid_BERT, inputs="text", outputs="text")
 covid_mental_io = Interface(fn=covid_MentalBERT, inputs="text", outputs="text")
 covid_phs_io = Interface(fn=covid_PHSBERT, inputs="text", outputs="text")
-covid = Parallel(covid_bert_io, covid_mental_io, covid_phs_io, 
+# covid = Parallel(covid_bert_io, covid_mental_io, covid_phs_io, 
+# 	examples=covid_examples, 
+# 	title=covid_title, 
+# 	description=covid_desc,
+# 	theme="peach")
+covid = Parallel(covid_bert_io, covid_phs_io, 
 	examples=covid_examples, 
 	title=covid_title, 
 	description=covid_desc,
@@ -198,7 +218,12 @@ def stress_PHSBERT(text):
 stress_bert_io = Interface(fn=stress_BERT, inputs="text", outputs="text")
 stress_mental_io = Interface(fn=stress_MentalBERT, inputs="text", outputs="text")
 stress_phs_io = Interface(fn=stress_PHSBERT, inputs="text", outputs="text")
-stress = Parallel(stress_bert_io, stress_mental_io, stress_phs_io, 
+# stress = Parallel(stress_bert_io, stress_mental_io, stress_phs_io, 
+# 	examples=stress_examples, 
+# 	title=stress_title, 
+# 	description=stress_desc,
+# 	theme="peach")
+stress = Parallel(stress_bert_io, stress_phs_io, 
 	examples=stress_examples, 
 	title=stress_title, 
 	description=stress_desc,
@@ -228,7 +253,12 @@ def other_PHSBERT(text):
 other_bert_io = Interface(fn=other_BERT, inputs="text", outputs="text")
 other_mental_io = Interface(fn=other_MentalBERT, inputs="text", outputs="text")
 other_phs_io = Interface(fn=other_PHSBERT, inputs="text", outputs="text")
-other = Parallel(other_bert_io, other_mental_io, other_phs_io, 
+# other = Parallel(other_bert_io, other_mental_io, other_phs_io, 
+# 	examples=other_examples, 
+# 	title=other_title, 
+# 	description=other_desc,
+# 	theme="peach")
+other = Parallel(other_bert_io, other_phs_io, 
 	examples=other_examples, 
 	title=other_title, 
 	description=other_desc,