Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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("
|
| 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,
|
|
|
|
|
|
|
| 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():
|