M

app.py

@@ -56,29 +56,152 @@ other_examples = [
 other_title = "Other Health Related Task"
 other_desc = "Enter drug-related tweets to generate labels from 3 models (BERT, MentalBERT, PHS-BERT). Label 0='no adverse drug reaction', 1='adverse drug reaction'. The three provided examples have true labels 0,0,1 respectively. 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")
-	predictor_mental = ktrain.load_predictor(folder + "mentalbert")
-	predictor_phs = ktrain.load_predictor(folder + "phsbert")
-
-	def BERT(text):
-		results = predictor_bert.predict(str(text))
-		return "BERT:" + str(results)
+# def make_interfaces(folder):
+# 	predictor_bert = ktrain.load_predictor(folder + "bert")
+# 	predictor_mental = ktrain.load_predictor(folder + "mentalbert")
+# 	predictor_phs = ktrain.load_predictor(folder + "phsbert")
+
+# 	def BERT(text):
+# 		results = predictor_bert.predict(str(text))
+# 		return "BERT:" + str(results)
 	
-	def MentalBERT(text):
-		results = predictor_mental.predict(str(text))
-		return "MentalBERT:" + str(results)
+# 	def MentalBERT(text):
+# 		results = predictor_mental.predict(str(text))
+# 		return "MentalBERT:" + str(results)
 		
-	def PHSBERT(text):
-		results = predictor_phs.predict(str(text))
-		return "PHS-BERT:" + str(results)
+# 	def PHSBERT(text):
+# 		results = predictor_phs.predict(str(text))
+# 		return "PHS-BERT:" + str(results)
+
+# 	bert_io = Interface(fn=BERT, inputs="text", outputs="text")
+# 	mental_io = Interface(fn=MentalBERT, inputs="text", outputs="text")
+# 	phs_io = Interface(fn=PHSBERT, inputs="text", outputs="text")
+
+# 	return bert_io, mental_io, phs_io
+folder = "vs/"
+vs_predictor_bert = ktrain.load_predictor(folder + "bert")
+vs_predictor_mental = ktrain.load_predictor(folder + "mentalbert")
+vs_predictor_phs = ktrain.load_predictor(folder + "phsbert")
+
+def vs_BERT(text):
+	results = vs_predictor_bert.predict(str(text))
+	return "BERT:" + str(results)
+
+def vs_MentalBERT(text):
+	results = vs_predictor_mental.predict(str(text))
+	return "MentalBERT:" + str(results)
+	
+def vs_PHSBERT(text):
+	results = vs_predictor_phs.predict(str(text))
+	return "PHS-BERT:" + str(results)
+
+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, 
+	examples=vs_examples, 
+	title=vs_title, 
+	description=vs_desc)
+
+folder = "hm/"
+hm_predictor_bert = ktrain.load_predictor(folder + "bert")
+hm_predictor_mental = ktrain.load_predictor(folder + "mentalbert")
+hm_predictor_phs = ktrain.load_predictor(folder + "phsbert")
+
+def hm_BERT(text):
+	results = hm_predictor_bert.predict(str(text))
+	return "BERT:" + str(results)
+
+def hm_MentalBERT(text):
+	results = hm_predictor_mental.predict(str(text))
+	return "MentalBERT:" + str(results)
+	
+def hm_PHSBERT(text):
+	results = hm_predictor_phs.predict(str(text))
+	return "PHS-BERT:" + 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)
+
+folder = "cv/"
+covid_predictor_bert = ktrain.load_predictor(folder + "bert")
+covid_predictor_mental = ktrain.load_predictor(folder + "mentalbert")
+covid_predictor_phs = ktrain.load_predictor(folder + "phsbert")
+
+def covid_BERT(text):
+	results = covid_predictor_bert.predict(str(text))
+	return "BERT:" + str(results)
+
+def covid_MentalBERT(text):
+	results = covid_predictor_mental.predict(str(text))
+	return "MentalBERT:" + str(results)
+	
+def covid_PHSBERT(text):
+	results = covid_predictor_phs.predict(str(text))
+	return "PHS-BERT:" + str(results)
+
+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, 
+	examples=covid_examples, 
+	title=covid_title, 
+	description=covid_desc)
+
+folder = "st/"
+stress_predictor_bert = ktrain.load_predictor(folder + "bert")
+stress_predictor_mental = ktrain.load_predictor(folder + "mentalbert")
+stress_predictor_phs = ktrain.load_predictor(folder + "phsbert")
+
+def stress_BERT(text):
+	results = stress_predictor_bert.predict(str(text))
+	return "BERT:" + str(results)
+
+def stress_MentalBERT(text):
+	results = stress_predictor_mental.predict(str(text))
+	return "MentalBERT:" + str(results)
+	
+def stress_PHSBERT(text):
+	results = stress_predictor_phs.predict(str(text))
+	return "PHS-BERT:" + str(results)
+
+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, 
+	examples=stress_examples, 
+	title=stress_title, 
+	description=stress_desc)
 
-	bert_io = Interface(fn=BERT, inputs="text", outputs="text")
-	mental_io = Interface(fn=MentalBERT, inputs="text", outputs="text")
-	phs_io = Interface(fn=PHSBERT, inputs="text", outputs="text")
+folder = "ot/"
+other_predictor_bert = ktrain.load_predictor(folder + "bert")
+other_predictor_mental = ktrain.load_predictor(folder + "mentalbert")
+other_predictor_phs = ktrain.load_predictor(folder + "phsbert")
 
-	return bert_io, mental_io, phs_io
+def other_BERT(text):
+	results = other_predictor_bert.predict(str(text))
+	return "BERT:" + str(results)
 
+def other_MentalBERT(text):
+	results = other_predictor_mental.predict(str(text))
+	return "MentalBERT:" + str(results)
+	
+def other_PHSBERT(text):
+	results = other_predictor_phs.predict(str(text))
+	return "PHS-BERT:" + str(results)
+
+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, 
+	examples=other_examples, 
+	title=other_title, 
+	description=other_desc)
 # vs_bert_io, vs_mental_io, vs_phs_io = make_interfaces("vs/")
 # vs = Parallel(vs_bert_io, vs_mental_io, vs_phs_io, 
 # 	examples=vs_examples, 
@@ -97,11 +220,11 @@ def make_interfaces(folder):
 # 	title=dep_title, 
 # 	description=dep_desc)
 
-covid_bert_io, covid_mental_io, covid_phs_io = make_interfaces("cv/")
-covid = Parallel(covid_bert_io, covid_mental_io, covid_phs_io, 
-	examples=covid_examples, 
-	title=covid_title, 
-	description=covid_desc)
+# covid_bert_io, covid_mental_io, covid_phs_io = make_interfaces("cv/")
+# covid = Parallel(covid_bert_io, covid_mental_io, covid_phs_io, 
+# 	examples=covid_examples, 
+# 	title=covid_title, 
+# 	description=covid_desc)
 
 # suicide_bert_io, suicide_mental_io, suicide_phs_io = make_interfaces("sc/")
 # suicide = Parallel(suicide_bert_io, suicide_mental_io, suicide_phs_io, 
@@ -109,11 +232,11 @@ covid = Parallel(covid_bert_io, covid_mental_io, covid_phs_io,
 # 	title=suicide_title, 
 # 	description=suicide_desc)
 
-stress_bert_io, stress_mental_io, stress_phs_io = make_interfaces("st/")
-stress = Parallel(stress_bert_io, stress_mental_io, stress_phs_io, 
-	examples=stress_examples, 
-	title=stress_title, 
-	description=stress_desc)
+# stress_bert_io, stress_mental_io, stress_phs_io = make_interfaces("st/")
+# stress = Parallel(stress_bert_io, stress_mental_io, stress_phs_io, 
+# 	examples=stress_examples, 
+# 	title=stress_title, 
+# 	description=stress_desc)
 
 # other_bert_io, other_mental_io, other_phs_io = make_interfaces("ot/")
 # other = Parallel(other_bert_io, other_mental_io, other_phs_io, 
@@ -121,10 +244,10 @@ stress = Parallel(stress_bert_io, stress_mental_io, stress_phs_io,
 # 	title=other_title, 
 # 	description=other_desc)
 
-desc = "Task is currently unavailable."
+# desc = "Task is currently unavailable."
 
-def model(text):
-	return "Predictions are currently unavailable."
+# def model(text):
+# 	return "Predictions are currently unavailable."
 
 # dep = Interface(fn=model, inputs="text", outputs="text", title=dep_title, description=desc)
 # covid = Interface(fn=model, inputs="text", outputs="text", title=covid_title, description=desc)
@@ -135,10 +258,10 @@ def model(text):
 # interfaces = [vs, hm, dep, covid, suicide, stress, other]
 # interface_names = [vs_title, hm_title, dep_title, covid_title, suicide_title, stress_title, other_title]
 
-# interfaces = [vs, hm, covid, stress, other]
-# interface_names = [vs_title, hm_title, covid_title, stress_title, other_title]
+interfaces = [vs, hm, covid, stress, other]
+interface_names = [vs_title, hm_title, covid_title, stress_title, other_title]
 
-interfaces = [covid, stress]
-interface_names = [covid_title, stress_title]
+# interfaces = [covid, stress]
+# interface_names = [covid_title, stress_title]
 
 TabbedInterface(interfaces, interface_names).launch()