set the variables

app.py

@@ -37,15 +37,19 @@ def predict_amputation(age, gender, race, diabetes_type):
 	#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": 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 "+diabetes_type+" diabetes"
+	input_dict = {"AGE": age, "GENDER": gender, "RACE": race, "DIABETES_CLASS":diabetes_class, "AMPUTATION":''}
 	
 	input_df = pd.DataFrame([input_dict])
+	
+	return str(predict(model=model, input_df=input_df))
 	#return str("ALLAH " + " " + str(age) + " " + gender + " " + race + diabetes_type)
-	return diabetes_type
+	#return diabetes_type
 	#return "ALLAH: "+str(predict(model=model, input_df=input_df)) # calls the predict function when 'submit' is clicked