M

app.py

@@ -84,16 +84,13 @@ 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)
+	return "BERT:" + str(vs_predictor_bert.predict(str(text)))
 
 def vs_MentalBERT(text):
-	results = vs_predictor_mental.predict(str(text))
-	return "MentalBERT:" + str(results)
+	return "MentalBERT:" + str(vs_predictor_mental.predict(str(text)))
 	
 def vs_PHSBERT(text):
-	results = vs_predictor_phs.predict(str(text))
-	return "PHS-BERT:" + str(results)
+	return "PHS-BERT:" + str(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")
@@ -109,16 +106,13 @@ 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)
+	return "BERT:" + str(hm_predictor_bert.predict(str(text)))
 
 def hm_MentalBERT(text):
-	results = hm_predictor_mental.predict(str(text))
-	return "MentalBERT:" + str(results)
+	return "MentalBERT:" + str(hm_predictor_mental.predict(str(text)))
 	
 def hm_PHSBERT(text):
-	results = hm_predictor_phs.predict(str(text))
-	return "PHS-BERT:" + str(results)
+	return "PHS-BERT:" + str(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,16 +128,13 @@ 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)
+	return "BERT:" + str(covid_predictor_bert.predict(str(text)))
 
 def covid_MentalBERT(text):
-	results = covid_predictor_mental.predict(str(text))
-	return "MentalBERT:" + str(results)
+	return "MentalBERT:" + str(covid_predictor_mental.predict(str(text)))
 	
 def covid_PHSBERT(text):
-	results = covid_predictor_phs.predict(str(text))
-	return "PHS-BERT:" + str(results)
+	return "PHS-BERT:" + str(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")
@@ -159,16 +150,13 @@ 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)
+	return "BERT:" + str(stress_predictor_bert.predict(str(text)))
 
 def stress_MentalBERT(text):
-	results = stress_predictor_mental.predict(str(text))
-	return "MentalBERT:" + str(results)
+	return "MentalBERT:" + str(stress_predictor_mental.predict(str(text)))
 	
 def stress_PHSBERT(text):
-	results = stress_predictor_phs.predict(str(text))
-	return "PHS-BERT:" + str(results)
+	return "PHS-BERT:" + str(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")
@@ -184,16 +172,13 @@ other_predictor_mental = ktrain.load_predictor(folder + "mentalbert")
 other_predictor_phs = ktrain.load_predictor(folder + "phsbert")
 
 def other_BERT(text):
-	results = other_predictor_bert.predict(str(text))
-	return "BERT:" + str(results)
+	return "BERT:" + str(other_predictor_bert.predict(str(text)))
 
 def other_MentalBERT(text):
-	results = other_predictor_mental.predict(str(text))
-	return "MentalBERT:" + str(results)
+	return "MentalBERT:" + str(other_predictor_mental.predict(str(text)))
 	
 def other_PHSBERT(text):
-	results = other_predictor_phs.predict(str(text))
-	return "PHS-BERT:" + str(results)
+	return "PHS-BERT:" + str(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")