Update app.py

app.py

@@ -61,14 +61,14 @@ def plot_digits(instances, images_per_row=3):
 
     n = len(instances)
 
-    fig = plt.figure(figsize=(30,15))
+    fig = plt.figure(figsize=(15,8))
     for i in range(len(instances)):
         # Debug, plot figure
         fig.add_subplot(n_rows, images_per_row, i + 1)
         #print(instances[i])
         plt.imshow(instances[i].reshape(size,size), cmap = mpl.cm.binary)
         plt.axis("off")
-        plt.title("Neighbor "+str(i+1), size=22)
+        plt.title("Neighbor "+str(i+1), size=20)
     fig.tight_layout()
 
     plt.savefig('results.png', dpi=300)
@@ -99,8 +99,8 @@ def KNN_predict(train_features, train_labels, test_feature, K):
   for k in range(K):
     major_class.append(sorted_labels[k][1])
 
-    # at most 60 neighbors for visualization
-    if k <60:
+    # at most 24 neighbors for visualization
+    if k <24:
       neighbor_feature = sorted_labels[k][2]
       neighbor_imgs.append(neighbor_feature)
   
@@ -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, 60, default=7)
+set_K = gr.inputs.Slider(0, 24, default=7)
 
 set_label = gr.outputs.Textbox(label="Predicted Digit")
 set_out_images = gr.outputs.Image(label="Closest Neighbors")