Update app.py

app.py

@@ -14,15 +14,15 @@ vs_predictor_phs = ktrain.load_predictor('vs/phsbert')
 
 def vs_BERT(text):
 	results = vs_predictor_bert.predict(str(text))
-	return str(results)
+	return "BERT:" + str(results)
 	
 def vs_MentalBERT(text):
 	results = vs_predictor_mental.predict(str(text))
-	return str(results)
+	return "MentalBERT:" + str(results)
 	
 def vs_PHSBERT(text):
 	results = vs_predictor_phs.predict(str(text))
-	return str(results)
+	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")
@@ -46,15 +46,15 @@ hm_predictor_phs = ktrain.load_predictor('hm/phsbert')
 
 def hm_BERT(text):
 	results = hm_predictor_bert.predict(str(text))
-	return str(results)
+	return "BERT:" + str(results)
 	
 def hm_MentalBERT(text):
 	results = hm_predictor_mental.predict(str(text))
-	return str(results)
+	return "MentalBERT" + str(results)
 	
 def hm_PHSBERT(text):
 	results = hm_predictor_phs.predict(str(text))
-	return str(results)
+	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")
@@ -65,22 +65,24 @@ hm = Parallel(hm_bert_io, hm_mental_io, hm_phs_io,
 	title=hm_title, 
 	description=hm_desc)
 	
+dep_title = "Depression Task"
+covid_title = "COVID Related Task"
+suicide_title = "Suicide Task"
+stress_title = "Stress Task"
+other_title = "Other Health Related Task"
+
+desc = "Task is currently unavailable."
+
 def model(text):
-	return "Predictions unavailable - to be completed."
+	return "Predictions are currently unavailable."
 
-dep = Interface(fn=model, inputs="text", outputs="text")
-covid = Interface(fn=model, inputs="text", outputs="text")
-suicide = Interface(fn=model, inputs="text", outputs="text")
-stress = Interface(fn=model, inputs="text", outputs="text")
-other = Interface(fn=model, inputs="text", outputs="text")
+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)
 
 interfaces = [vs, hm, dep, covid, suicide, stress, other]
-interface_names = ["Vaccine Sentiment Task", 
-	"Health Mention Task",
-	"Depression Task",
-	"COVID Related Task",
-	"Suicide Task",
-	"Stress Task",
-	"Other Health Related Task"]
+interface_names = [vs_title, hm_title, dep_title, covid_title, suicide_title, stress_title, other_title]
 
 TabbedInterface(interfaces, interface_names).launch()