Update app.py

app.py

@@ -50,6 +50,7 @@ def gradcam(img_create):
   fig,ax = plt.subplots()
   x_dec.show(ctx=ax)
   ax.imshow(cam_map[1].detach().cpu(), alpha=0.6, extent=(0,128,128,0), interpolation='bilinear', cmap='magma');
+  fig.savefig("gcam.jpg")
 
   return dict(zip(categories, map(float, probs))), fig
 
@@ -69,6 +70,6 @@ examples=["filibe-1-1.jpg",
           "varna-1-1.jpg"]
 
 
-demo = gr.Interface(fn=gradcam, inputs=image, outputs=[label, gr.inputs.Image(shape=(128,128))], examples=examples)
+demo = gr.Interface(fn=gradcam, inputs=image, outputs=[label, "image"], examples=examples)
 
 demo.launch(inline=False)