Update app.py

app.py

@@ -9,7 +9,20 @@ from PIL import Image
 
 import clip
 from dalle.models import Dalle
-from dalle.utils.utils import clip_score
+from dalle.utils.utils import clip_score, download
+
+url = "https://arena.kakaocdn.net/brainrepo/models/minDALL-E/57b008f02ceaa02b779c8b7463143315/1.3B.tar.gz"
+root = os.path.expanduser("~/.cache/minDALL-E")
+filename = os.path.basename(url)
+pathname = filename[:-len('.tar.gz')]
+
+expected_md5 = url.split("/")[-2]
+download_target = os.path.join(root, filename)
+result_path = os.path.join(root, pathname)
+
+if not os.path.exists(result_path):
+    result_path = download(url, root)
+
 
 
 device = "cpu"