cleaned code and set output for amputation risk

app.py

@@ -17,42 +17,23 @@ def predict(model, input_df):
 	predictions_df = predict_model(estimator=model, data=input_df)
 	predict_label = predictions_df["Label"][0] # either 1 (amputation yes) or 0 (amputation no)
 	predict_score = predictions_df["Score"][0] # the prediction (accuracy)
-	return "<b>AMPUTATION RISK: " + str(predict_label) + " SCORE: "+str(predict_score)+"</b>"  
-
-
-
-
-#input_dict = {"AGE": age, "GENDER_F": gender, "RACE_Asian":	,"RACE_Black": ,	"RACE_Coloured":,	"RACE_Other":,	"RACE_White":,	"DIABETES_CLASS_Type 1 diabetes":}
-
-
-
-
+	amputation_risk = ""
+	if predict_label == 1:
+		amputation_risk = "YES"
+	else:
+		amputation_risk = "NO"
+	return "AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)  
 
 
 # the parameters in this function, actually gets the inputs for the prediction 
 def predict_amputation(age, gender, race, diabetes_type):
-	#input_dict = {"AGE": 70.0, "GENDER_F": 0.0, "RACE_Asian": 1.0, "RACE_Black": 0.0,	"RACE_Coloured": 0.0,	"RACE_Other": 0.0,	"RACE_White": 0.0,	"DIABETES_CLASS_Type 1 diabetes":0.0}
-	
-	
-	#input_dict = {"AGE": 70.0, "GENDER": 0.0, "RACE": 1.0, "DIABETES_CLASS":0.0, "AMPUTATION":0}
-	
-	
-	#input_dict = {"AGE": 70, "GENDER": "F", "RACE": "Asian", "DIABETES_CLASS":"Type 2 diabetes", "AMPUTATION":''}
-	
-	
-	#input_dict = {"AGE": 80, "GENDER": "F", "RACE": "Asian", "DIABETES_CLASS":"Type 2 diabetes", "AMPUTATION":''}
-	
 	diabetes_class = "Type "+str(diabetes_type)+" diabetes"
 	gender = gender[0]
 	input_dict = {"AGE": age, "GENDER": gender, "RACE": race, "DIABETES_CLASS":diabetes_class, "AMPUTATION":''}
 	
 	input_df = pd.DataFrame([input_dict])
 	
-	return gender#str(predict(model=model, input_df=input_df))
-	#return str("ALLAH " + " " + str(age) + " " + gender + " " + race + diabetes_type)
-	#return diabetes_type
-	#return "ALLAH: "+str(predict(model=model, input_df=input_df)) # calls the predict function when 'submit' is clicked
-
+	return str(predict(model=model, input_df=input_df)) # calls the predict function when 'submit' is clicked
 
 
 
@@ -63,8 +44,6 @@ description = "A diabetes-related amputation machine learning model trained on t
 article = "<p style='text-align: center'><span style='font-size: 15pt;'>Copyright &copy; DIARC. 2021. All Rights Reserved. Contact Us: <a href='mailto:smtshali@wol.co.za'>Dr Sifisiso Mtshali</a> or <a href='mailto:mahomedo@ukzn.ac.za'>Dr Ozayr Mahomed</a></span></p>"
 
 
-
-
 iface = gr.Interface(
 		fn=predict_amputation, 
 		title=title, 
@@ -82,6 +61,7 @@ iface = gr.Interface(
 		],
 )
 
+
 iface.test_launch()		
 if __name__ == "__main__":
 	iface.launch()