MohammedAH commited on
Commit
069fc47
·
verified ·
1 Parent(s): 28428d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- file_name = "brain_mri.h5"
36
  )
37
 
38
  segmentation_path = hf_hub_download(
39
  repo_id = "MohammedAH/Unet-Brain-segmentation",
40
- file_name = "Unet_model.h5"
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