change output

app.py

@@ -19,9 +19,13 @@ def predict(model, input_df):
 	amputation_risk = ""
 	if predict_label == 1:
 		amputation_risk = "YES"
+		output = "<div style='background-color:green;'>"+"AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)+"</div>"
 	else:
 		amputation_risk = "NO"
-	return "AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)  
+		output = "<div style='background-color:green;'>"+"AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)+"</div>"
+	
+	
+	return output#"AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)  
 
 
 # the parameters in this function, actually gets the inputs for the prediction 
@@ -32,6 +36,7 @@ def predict_amputation(age, gender, race, diabetes_type):
 	
 	input_df = pd.DataFrame([input_dict])
 	
+	# output 
 	return str(predict(model=model, input_df=input_df)) # calls the predict function when 'submit' is clicked
 
 
@@ -60,7 +65,6 @@ iface = gr.Interface(
 			[30, "Male", "Coloured", 1],
 			[65, "Female", "Other", 2],
 		],
-		outputs.HighlightedText(color_map={"+": "lightgreen", "-": "pink", })
 )