Ahsen Khaliq commited on
Commit ·
ce94770
1
Parent(s): bf07612
Update app.py
Browse files
app.py
CHANGED
|
@@ -77,7 +77,9 @@ def run_one_image(img, model):
|
|
| 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")
|
|
|
|
| 77 |
|
| 78 |
plt.subplot(1, 4, 4)
|
| 79 |
show_image(im_paste[0], "reconstruction + visible")
|
| 80 |
+
|
| 81 |
plt.savefig("test.png",bbox_inches='tight')
|
| 82 |
+
|
| 83 |
|
| 84 |
# download checkpoint if not exist
|
| 85 |
os.system("wget -nc https://dl.fbaipublicfiles.com/mae/visualize/mae_visualize_vit_large.pth")
|