Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
d21c058
1
Parent(s):
62e71f5
Updated app.py
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ import torch
|
|
| 7 |
import nibabel as nib
|
| 8 |
import matplotlib.pyplot as plt
|
| 9 |
import spaces # Import spaces for GPU decoration
|
|
|
|
| 10 |
|
| 11 |
# Define paths
|
| 12 |
MODEL_DIR = "./model" # Local directory to store the downloaded model
|
|
|
|
| 7 |
import nibabel as nib
|
| 8 |
import matplotlib.pyplot as plt
|
| 9 |
import spaces # Import spaces for GPU decoration
|
| 10 |
+
import numpy as np
|
| 11 |
|
| 12 |
# Define paths
|
| 13 |
MODEL_DIR = "./model" # Local directory to store the downloaded model
|