Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,12 +32,12 @@ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
| 32 |
def get_model_path():
|
| 33 |
classification_path = hf_hub_download(
|
| 34 |
repo_id = "MohammedAH/Brrain-MRI-Classsification",
|
| 35 |
-
|
| 36 |
)
|
| 37 |
|
| 38 |
segmentation_path = hf_hub_download(
|
| 39 |
repo_id = "MohammedAH/Unet-Brain-segmentation",
|
| 40 |
-
|
| 41 |
)
|
| 42 |
|
| 43 |
return classification_path, segmentation_path
|
|
|
|
| 32 |
def get_model_path():
|
| 33 |
classification_path = hf_hub_download(
|
| 34 |
repo_id = "MohammedAH/Brrain-MRI-Classsification",
|
| 35 |
+
filename = "brain_mri.h5"
|
| 36 |
)
|
| 37 |
|
| 38 |
segmentation_path = hf_hub_download(
|
| 39 |
repo_id = "MohammedAH/Unet-Brain-segmentation",
|
| 40 |
+
filename = "Unet_model.h5"
|
| 41 |
)
|
| 42 |
|
| 43 |
return classification_path, segmentation_path
|