Update app.py

app.py

@@ -17,7 +17,7 @@ input_module2 = gr.inputs.Slider(1, 10, step=1, label = "k")
 
 output_module1 = gr.outputs.Textbox(label = "Predicted Digit")
 
-output_module2 = gr.outputs.Image(label = "Predicted Probability per class")
+output_module2 = gr.outputs.Label(label = "Predicted Probability per class" num_top_classes=4)
 
 def image_classification(input1, input2):
     image = input1.reshape(1, 28 *28)
@@ -30,7 +30,7 @@ def image_classification(input1, input2):
     y_test_predicted_label = kNN_classifier.predict(image)
 
     output1 = y_test_predicted_label[0]# text output example
-    output2 = y_test_predicted_label[1] # image-like array output example
+    output2 = y_test_predicted_label # image-like array output example
     return output1,output2
 
 # Step 6.4: Put all three component together into the gradio's interface function