Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ model_path = "unet_model.h5"
|
|
15 |
|
16 |
# Safe download with streaming to avoid incomplete file
|
17 |
if not os.path.exists(model_path):
|
18 |
-
hf_url = "https://huggingface.co/rishirajbal/UNET_plus_plus_Brain_segmentation"
|
19 |
print(f"Downloading model from {hf_url}...")
|
20 |
with requests.get(hf_url, stream=True) as r:
|
21 |
r.raise_for_status()
|
|
|
15 |
|
16 |
# Safe download with streaming to avoid incomplete file
|
17 |
if not os.path.exists(model_path):
|
18 |
+
hf_url = "https://huggingface.co/rishirajbal/UNET_plus_plus_Brain_segmentation/resolve/main/unet_model.h5"
|
19 |
print(f"Downloading model from {hf_url}...")
|
20 |
with requests.get(hf_url, stream=True) as r:
|
21 |
r.raise_for_status()
|