Update app.py

app.py

@@ -49,8 +49,8 @@ def gradcam(img_create):
   x_dec = TensorImage(dls_augmented.train.decode((x,))[0][0])
   fig,ax = plt.subplots()
   x_dec.show(ctx=ax)
-  ax.imshow(cam_map[1].detach().cpu(), alpha=0.6, extent=(0,135,135,0), interpolation='bilinear', cmap='magma');
-  fig.savefig("gcam.jpg")
+  ax.imshow(cam_map[1].detach().cpu(), alpha=0.6, extent=(0,128,128,0), interpolation='bilinear', cmap='magma');
+  fig.savefig("gcam.jpg", dpi=199)
   im=cv2.imread("gcam.jpg",1)
   im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
   #im=np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
@@ -73,6 +73,6 @@ examples=["filibe-1-1.jpg",
           "varna-1-1.jpg"]
 
 
-demo = gr.Interface(fn=gradcam, inputs=image, outputs=[gr.inputs.Image(shape=(135,135)), label], examples=examples)
+demo = gr.Interface(fn=gradcam, inputs=image, outputs=[gr.inputs.Image(shape=(128,128)), label], examples=examples)
 
 demo.launch(inline=False)