Update app.py

app.py

@@ -52,9 +52,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');
-  #fig.savefig("gcam.jpg")
-  #im=cv2.imread("gcam.jpg",1)
-  im=np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
+  fig.savefig("gcam.jpg")
+  im=cv2.imread("gcam.jpg",1)
+  im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
+  #im=np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
 
   return dict(zip(categories, map(float, probs))), im
 
@@ -74,6 +75,6 @@ examples=["filibe-1-1.jpg",
           "varna-1-1.jpg"]
 
 
-demo = gr.Interface(fn=gradcam, inputs=image, outputs=[label, "image"], examples=examples)
+demo = gr.Interface(fn=gradcam, inputs=image, outputs=[label, gr.inputs.Image(shape=(128,128))], examples=examples)
 
 demo.launch(inline=False)