Update app.py

app.py

@@ -42,7 +42,10 @@ 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');
-    return (dict(zip(categories, map(float, probs))), fig)
+    im="gcam.jpg"
+    fig.savefig(im)
+    
+    return (dict(zip(categories, map(float, probs))), im)
     
 
 image=gr.inputs.Image(shape=(128,128))