Update app.py

app.py

@@ -18,10 +18,6 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-import sys
-
-sys.path.append('../../src')
-
 import argparse
 import random
 import time
@@ -29,6 +25,7 @@ import json
 import os
 import glob
 import pathlib
+import requests
 from functools import partial
 from pprint import pprint
 from multiprocessing.connection import Client, Listener
@@ -95,6 +92,19 @@ opt = parser.parse_args()
 
 device = f'cuda:{opt.device}' if opt.device >= 0 else 'cpu'
 
+def download_file(url, root):
+    local_filename = url.split('/')[-1]
+    # NOTE the stream=True parameter below
+    with requests.get(url, stream=True) as r:
+        r.raise_for_status()
+        with open(os.path.join(root, local_filename), 'wb') as f:
+            for chunk in r.iter_content(chunk_size=8192): 
+                # If you have chunk encoded response uncomment if
+                # and set chunk_size parameter to None.
+                #if chunk: 
+                f.write(chunk)
+    return local_filename
+
 
 if opt.model is None:
     # opt.model = 'cagliostrolab/animagine-xl-3.1'
@@ -105,6 +115,11 @@ else:
     if opt.model.endswith('.safetensors'):
         opt.model = os.path.abspath(os.path.join('checkpoints', opt.model))
 
+root = pathlib.Path(__file__).parent
+checkpoint_root = os.path.join(root, 'checkpoints')
+opt.model = download_file(opt.model, checkpoint_root)
+print(opt.model, checkpoint_root)
+        
 # model = StreamMultiDiffusionSDXL(
 model = StreamMultiDiffusion(
     device,
@@ -160,7 +175,6 @@ opt.prep_time = -3
 
 ### Load examples
 
-root = pathlib.Path(__file__).parent
 example_root = os.path.join(root, 'examples')
 example_images = glob.glob(os.path.join(example_root, '*.png'))
 example_images = [Image.open(i) for i in example_images]