Ahsen Khaliq
commited on
Commit
·
bf07612
1
Parent(s):
8165779
Update app.py
Browse files
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")
|