Update app.py

app.py

@@ -91,7 +91,7 @@ demo = gr.Interface(
   fn = gen_grad_img_single,
   inputs = gr.Image(type = "pil", shape = (224,224)),
   # outputs = [gr.outputs.Label(num_top_classes = 2, label = 'Classifiaction'), gr.Textbox('infer_time', label = 'Inference Time(ms)')]
-  outputs = ["image", gr.Textbox('y_pred', label = 'Prediction')]
+  outputs = [gr.Image(type = "numpy"), gr.Textbox('y_pred', label = 'Prediction')]
    )
 demo.launch()