Spaces:

pytorch
/

FCN

Runtime error

akhaliq HF Staff commited on Feb 23, 2022

Commit

c06152d

1 Parent(s): 9a34825

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -42,7 +42,8 @@ def inference(input_image):
     r = Image.fromarray(output_predictions.byte().cpu().numpy()).resize(input_image.size)
     r.putpalette(colors)
     plt.imshow(r)
-    plt.savefig('out.png')
     return 'out.png'

     r = Image.fromarray(output_predictions.byte().cpu().numpy()).resize(input_image.size)
     r.putpalette(colors)
     plt.imshow(r)
+    plt.axis('off')
+    plt.savefig('out.png',bbox_inches='tight')
     return 'out.png'