Update app.py

app.py

@@ -17,8 +17,6 @@ input_module2 = gr.inputs.Slider(1, 10, step=1, label = "k")
 
 output_module1 = gr.outputs.Textbox(label = "Predicted Digit")
 
-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)
     X_train = x_train.reshape(60000, 28*28)
@@ -30,11 +28,10 @@ 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 # image-like array output example
-    return output1,output2
-
+    
+    return output1
 # Step 6.4: Put all three component together into the gradio's interface function 
 gr.Interface(fn=image_classification, 
              inputs=[input_module1, input_module2], 
-             outputs=[output_module1, output_module2]
+             outputs=[output_module1]
             ).launch(debug = True)