Update app.py

app.py

@@ -116,7 +116,7 @@ def KNN_predict(train_features, train_labels, test_feature, K):
 ### main function for gradio to call to classify image
 def call_our_KNN(test_image, K=7):
   test_image_flatten = test_image.reshape((-1, 28*28))
-  y_pred_each, image_path = KNN_predict(train_features, train_labels, test_image_flatten, K)
+  y_pred_each, image_path = KNN_predict(train_features, train_labels, test_image_flatten, int(K))
   return y_pred_each, image_path
 
 
@@ -125,7 +125,7 @@ sample_images = get_sample_images(10)
 
 ### configure inputs/outputs
 set_image = gr.inputs.Image(shape=(28, 28), image_mode='L')
-set_K = gr.inputs.Slider(0, 24, default=7)
+set_K = gr.inputs.Slider(0, 24, step=1, default=7)
 
 set_label = gr.outputs.Textbox(label="Predicted Digit")
 set_out_images = gr.outputs.Image(label="Closest Neighbors")