Ahsen Khaliq commited on
Commit
bf07612
1 Parent(s): 8165779

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -77,10 +77,7 @@ def run_one_image(img, model):
77
 
78
  plt.subplot(1, 4, 4)
79
  show_image(im_paste[0], "reconstruction + visible")
80
-
81
- plt.show()
82
  plt.savefig("test.png",bbox_inches='tight')
83
-
84
 
85
  # download checkpoint if not exist
86
  os.system("wget -nc https://dl.fbaipublicfiles.com/mae/visualize/mae_visualize_vit_large.pth")
 
77
 
78
  plt.subplot(1, 4, 4)
79
  show_image(im_paste[0], "reconstruction + visible")
 
 
80
  plt.savefig("test.png",bbox_inches='tight')
 
81
 
82
  # download checkpoint if not exist
83
  os.system("wget -nc https://dl.fbaipublicfiles.com/mae/visualize/mae_visualize_vit_large.pth")