interface

app.py

@@ -56,26 +56,6 @@ 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)."
 
-# vs_predictor_bert = ktrain.load_predictor('vs/bert')
-# vs_predictor_mental = ktrain.load_predictor('vs/mentalbert')
-# vs_predictor_phs = ktrain.load_predictor('vs/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")
-
 def make_interfaces(folder):
 	predictor_bert = ktrain.load_predictor(folder + "bert")
 	predictor_mental = ktrain.load_predictor(folder + "mentalbert")
@@ -100,50 +80,46 @@ def make_interfaces(folder):
 	return bert_io, mental_io, phs_io
 
 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, 
 	title=vs_title, 
 	description=vs_desc)
 
-# 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 "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_bert_io, hm_mental_io, hm_phs_io = make_interfaces("hm/")
-
 hm = Parallel(hm_bert_io, hm_mental_io, hm_phs_io, 
 	examples=hm_examples, 
 	title=hm_title, 
 	description=hm_desc)
 	
-
-desc = "Task is 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)
-suicide = Interface(fn=model, inputs="text", outputs="text", title=suicide_title, description=desc)
-stress = Interface(fn=model, inputs="text", outputs="text", title=stress_title, description=desc)
-other = Interface(fn=model, inputs="text", outputs="text", title=other_title, description=desc)
+dep_bert_io, dep_mental_io, dep_phs_io = make_interfaces("dp/")
+dep = Parallel(dep_bert_io, dep_mental_io, dep_phs_io, 
+	examples=dep_examples, 
+	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)
+
+suicide_bert_io, suicide_mental_io, suicide_phs_io = make_interfaces("sc/")
+suicide = Parallel(suicide_bert_io, suicide_mental_io, suicide_phs_io, 
+	examples=suicide_examples, 
+	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)
+
+other_bert_io, other_mental_io, other_phs_io = make_interfaces("ot/")
+other = Parallel(other_bert_io, other_mental_io, other_phs_io, 
+	examples=other_examples, 
+	title=other_title, 
+	description=other_desc)
 
 interfaces = [vs, hm, dep, covid, suicide, stress, other]
 interface_names = [vs_title, hm_title, dep_title, covid_title, suicide_title, stress_title, other_title]