Update app.py

app.py

@@ -82,14 +82,22 @@ 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_output(number):
+	if int(number) == 0:
+		return "vaccine critical"
+	if int(number) == 1:
+		return "neutral"
+	if int(number) == 2:
+		return "vaccine supportive"
+
 def vs_BERT(text):
-	return "BERT:" + str(vs_predictor_bert.predict(str(text)))
+	return "BERT ➡" + vs_output(vs_predictor_bert.predict(str(text)))
 
 def vs_MentalBERT(text):
-	return "MentalBERT:" + str(vs_predictor_mental.predict(str(text)))
+	return "MentalBERT ➡" + vs_output(vs_predictor_mental.predict(str(text)))
 	
 def vs_PHSBERT(text):
-	return "PHS-BERT:" + str(vs_predictor_phs.predict(str(text)))
+	return "PHS-BERT ➡" + vs_output(vs_predictor_phs.predict(str(text)))
 
 vs_bert_io = Interface(fn=vs_BERT, inputs="text", outputs="text")
 vs_mental_io = Interface(fn=vs_MentalBERT, inputs="text", outputs="text")