Update app.py

app.py

@@ -55,7 +55,9 @@ def gradcam(img):
     ax.imshow(cam_map[1].detach().cpu(), alpha=0.6, extent=(0,128,128,0), interpolation='bilinear', cmap='magma');
     return gcam
 
-demo = gr.Interface(sepia, gr.Image(shape=(200, 200)), "image")
-demo = gr.Interface(fn=classify_img, inputs=image, outputs=label, examples=examples)
+
+
+demo = gr.Interface(fn=gradcam, inputs=image, outputs="image", examples=examples)
+#demo = gr.Interface(fn=classify_img, inputs=image, outputs=label, examples=examples)
 
 demo.launch(inline=False)