Update app.py

app.py

@@ -93,7 +93,7 @@ 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]
+    local_filename = url.split('/')[-1].split('?')[0]
     # NOTE the stream=True parameter below
     with requests.get(url, stream=True) as r:
         r.raise_for_status()
@@ -103,22 +103,22 @@ def download_file(url, root):
                 # and set chunk_size parameter to None.
                 #if chunk: 
                 f.write(chunk)
-    return local_filename
+    return os.path.join(root, local_filename)
 
 
 if opt.model is None:
     # opt.model = 'cagliostrolab/animagine-xl-3.1'
     # opt.model = 'ironjr/BlazingDriveV11m'
     # opt.model = 'KBlueLeaf/kohaku-v2.1'
-    opt.model = 'https://huggingface.co/ironjr/BlazingDriveV13md/raw/main/blazingDrive_V13md.safetensors'
+    opt.model = 'https://huggingface.co/ironjr/BlazingDriveV13md/resolve/main/blazingDrive_V13md.safetensors?download=true'
+    
+    root = pathlib.Path(__file__).parent
+    checkpoint_root = os.path.join(root, 'checkpoints')
+    opt.model = download_file(opt.model, checkpoint_root)
+    print(opt.model, os.path.getsize(opt.model))
 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(