Update app.py

app.py

@@ -52,8 +52,9 @@ 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")
-  im=cv2.imread("gcam.jpg",1)
+  #fig.savefig("gcam.jpg")
+  #im=cv2.imread("gcam.jpg",1)
+  im=np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
 
   return dict(zip(categories, map(float, probs))), im