Update app.py

app.py

@@ -5,8 +5,8 @@ vs_examples = [
 	["I only get my kids the ones I got....I've turned down many so called 'vaccines'"], 
 	["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 - VS2"
-vs_desc = "Enter vaccine-related tweets to generate sentiment from 3 models (BERT, MentalBERT, PHS-BERT). Label 0='vaccine critical', 1='neutral', 2='vaccine supportive'. The three provided examples have true labels 0,1,2 respectively. For details about VS2, please refer to our paper (linked provided in the corresponding Hugging Face repository)."
+vs_title = "Vaccine Sentiment Task"
+vs_desc = "Enter vaccine-related tweets to generate sentiment from 3 models (BERT, MentalBERT, PHS-BERT). Label 0='vaccine critical', 1='neutral', 2='vaccine supportive'. The three provided examples have true labels 0,1,2 respectively. For more details, please refer to the VS2 description in our paper (linked provided in the corresponding Hugging Face repository)."
 
 vs_predictor_bert = ktrain.load_predictor('vs/bert')
 vs_predictor_mental = ktrain.load_predictor('vs/mentalbert')
@@ -31,13 +31,43 @@ vs_phs_io = Interface(fn=vs_PHSBERT, inputs="text", outputs="text")
 vs = Parallel(vs_bert_io, vs_mental_io, vs_phs_io, 
 	examples=vs_examples, 
 	title=vs_title, 
-	description=vs_desc,
-	theme="peach")
+	description=vs_desc)
+	
+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?"], 
+	["It's odd how humans are considered predators when they have a heart attack and run like mad when a small insect is running straight towards them."], 
+	["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."]]
+hm_title = "Health Mention Task"
+hm_desc = "Enter health-related tweets to generate classification from 3 models (BERT, MentalBERT, PHS-BERT). Label 0='', 1='', 2=''. The three provided examples have true labels 0,1,2 respectively. For more details, please refer to the X description in our paper (linked provided in the corresponding Hugging Face repository)."
+
+hm_predictor_bert = ktrain.load_predictor('hm/bert')
+hm_predictor_mental = ktrain.load_predictor('hm/mentalbert')
+hm_predictor_phs = ktrain.load_predictor('hm/phsbert')
+
+def hm_BERT(text):
+	results = hm_predictor_bert.predict(str(text))
+	return str(results)
+	
+def hm_MentalBERT(text):
+	results = hm_predictor_mental.predict(str(text))
+	return str(results)
+	
+def hm_PHSBERT(text):
+	results = hm_predictor_phs.predict(str(text))
+	return str(results)
+
+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, 
+	examples=hm_examples, 
+	title=hm_title, 
+	description=hm_desc)
 	
 def model(text):
 	return "Predictions unavailable - to be completed."
 
-hm = Interface(fn=model, inputs="text", outputs="text")
 dep = Interface(fn=model, inputs="text", outputs="text")
 covid = Interface(fn=model, inputs="text", outputs="text")
 suicide = Interface(fn=model, inputs="text", outputs="text")