Update app.py

app.py

@@ -89,6 +89,7 @@ def vs_output(number):
 		return "neutral"
 	if int(number) == 2:
 		return "vaccine supportive"
+	return "N/A"
 
 def vs_BERT(text):
 	return "BERT ➡" + vs_output(vs_predictor_bert.predict(str(text)))
@@ -112,14 +113,23 @@ 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_output(number):
+	if int(number) == 0:
+		return "Figurative Health Mentions"
+	if int(number) == 1:
+		return "Non-Health Mentions"
+	if int(number) == 2:
+		return "Health Mentions"
+	return "N/A"
+
 def hm_BERT(text):
-	return "BERT:" + str(hm_predictor_bert.predict(str(text)))
+	return "BERT ➡ " + hm_output(hm_predictor_bert.predict(str(text)))
 
 def hm_MentalBERT(text):
-	return "MentalBERT:" + str(hm_predictor_mental.predict(str(text)))
+	return "MentalBERT ➡ " + hm_output(hm_predictor_mental.predict(str(text)))
 	
 def hm_PHSBERT(text):
-	return "PHS-BERT:" + str(hm_predictor_phs.predict(str(text)))
+	return "PHS-BERT ➡ " + hm_output(hm_predictor_phs.predict(str(text)))
 
 hm_bert_io = Interface(fn=hm_BERT, inputs="text", outputs="text")
 hm_mental_io = Interface(fn=hm_MentalBERT, inputs="text", outputs="text")
@@ -134,14 +144,21 @@ 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 cv_output(number):
+	if int(number) == 0:
+		return "news"
+	if int(number) == 1:
+		return "personal narrative"
+	return "N/A"
+
 def covid_BERT(text):
-	return "BERT:" + str(covid_predictor_bert.predict(str(text)))
+	return "BERT ➡ " + cv_output(covid_predictor_bert.predict(str(text)))
 
 def covid_MentalBERT(text):
-	return "MentalBERT:" + str(covid_predictor_mental.predict(str(text)))
+	return "MentalBERT ➡ " + cv_output(covid_predictor_mental.predict(str(text)))
 	
 def covid_PHSBERT(text):
-	return "PHS-BERT:" + str(covid_predictor_phs.predict(str(text)))
+	return "PHS-BERT ➡ " + cv_output(covid_predictor_phs.predict(str(text)))
 
 covid_bert_io = Interface(fn=covid_BERT, inputs="text", outputs="text")
 covid_mental_io = Interface(fn=covid_MentalBERT, inputs="text", outputs="text")
@@ -156,14 +173,21 @@ 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 st_output(number):
+	if int(number) == 0:
+		return "no stress"
+	if int(number) == 1:
+		return "stress"
+	return "N/A"
+
 def stress_BERT(text):
-	return "BERT:" + str(stress_predictor_bert.predict(str(text)))
+	return "BERT ➡ " + st_output(stress_predictor_bert.predict(str(text)))
 
 def stress_MentalBERT(text):
-	return "MentalBERT:" + str(stress_predictor_mental.predict(str(text)))
+	return "MentalBERT ➡ " + st_output(stress_predictor_mental.predict(str(text)))
 	
 def stress_PHSBERT(text):
-	return "PHS-BERT:" + str(stress_predictor_phs.predict(str(text)))
+	return "PHS-BERT ➡ " + st_output(stress_predictor_phs.predict(str(text)))
 
 stress_bert_io = Interface(fn=stress_BERT, inputs="text", outputs="text")
 stress_mental_io = Interface(fn=stress_MentalBERT, inputs="text", outputs="text")
@@ -178,14 +202,21 @@ other_predictor_bert = ktrain.load_predictor(folder + "bert")
 other_predictor_mental = ktrain.load_predictor(folder + "mentalbert")
 other_predictor_phs = ktrain.load_predictor(folder + "phsbert")
 
+def ot_output(number):
+	if int(number) == 0:
+		return "no adverse drug reaction"
+	if int(number) == 1:
+		return "adverse drug reaction"
+	return "N/A"
+
 def other_BERT(text):
-	return "BERT:" + str(other_predictor_bert.predict(str(text)))
+	return "BERT ➡ " + ot_output(other_predictor_bert.predict(str(text)))
 
 def other_MentalBERT(text):
-	return "MentalBERT:" + str(other_predictor_mental.predict(str(text)))
+	return "MentalBERT ➡ " + ot_output(other_predictor_mental.predict(str(text)))
 	
 def other_PHSBERT(text):
-	return "PHS-BERT:" + str(other_predictor_phs.predict(str(text)))
+	return "PHS-BERT ➡ " + ot_output(other_predictor_phs.predict(str(text)))
 
 other_bert_io = Interface(fn=other_BERT, inputs="text", outputs="text")
 other_mental_io = Interface(fn=other_MentalBERT, inputs="text", outputs="text")