FrancescoLR commited on
Commit
5456f4c
·
verified ·
1 Parent(s): 7d444b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -28,7 +28,7 @@ def download_model():
28
  print("Downloading BrainAgeNeXt model weights...")
29
  for i in range(1, 6):
30
  hf_hub_download(repo_id=REPO_ID, filename=f"BrainAge_{i}.pth", cache_dir=MODEL_DIR)
31
- print("✅ BrainAgeNeXt model downloaded successfully.")
32
 
33
  # 🔹 Function to Load Model
34
  def initialize_model():
@@ -152,7 +152,9 @@ with gr.Blocks() as demo:
152
  1. La Rosa, F., Dos Santos Silva, J., Dereskewicz, E., Invernizzi, A., Cahan, N., Galasso, J., ... & Beck, E. S. (2025).
153
  BrainAgeNeXt: Advancing Brain Age Modeling for Individuals with Multiple Sclerosis. Imaging Neuroscience.
154
  DOI: [10.1162/imag_a_00487](https://doi.org/10.1162/imag_a_00487)
155
- 2. Roy, Saikat, et al. "Mednext: transformer-driven scaling of convnets for medical image segmentation." MICCAI, 2023.
 
 
156
  """)
157
  # 🔹 Debugging GPU Environment
158
  if torch.cuda.is_available():
 
28
  print("Downloading BrainAgeNeXt model weights...")
29
  for i in range(1, 6):
30
  hf_hub_download(repo_id=REPO_ID, filename=f"BrainAge_{i}.pth", cache_dir=MODEL_DIR)
31
+ print("BrainAgeNeXt model downloaded successfully.")
32
 
33
  # 🔹 Function to Load Model
34
  def initialize_model():
 
152
  1. La Rosa, F., Dos Santos Silva, J., Dereskewicz, E., Invernizzi, A., Cahan, N., Galasso, J., ... & Beck, E. S. (2025).
153
  BrainAgeNeXt: Advancing Brain Age Modeling for Individuals with Multiple Sclerosis. Imaging Neuroscience.
154
  DOI: [10.1162/imag_a_00487](https://doi.org/10.1162/imag_a_00487)
155
+ 2. Roy, S., Koehler, G., Ulrich, C., Baumgartner, M., Petersen, J., Isensee, F., Jaeger, P.F. & Maier-Hein, K. (2023).
156
+ MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentation.
157
+ International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI).
158
  """)
159
  # 🔹 Debugging GPU Environment
160
  if torch.cuda.is_available():