Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ class EfficientNetClassifier(nn.Module):
|
|
| 60 |
|
| 61 |
# ββ Load model (cached after first download) ββββββββββββββββββββββ
|
| 62 |
def load_model():
|
| 63 |
-
ckpt_path = hf_hub_download(repo_id=
|
| 64 |
ckpt = torch.load(ckpt_path, map_location=DEVICE, weights_only=False)
|
| 65 |
|
| 66 |
n_classes = ckpt.get("num_classes", 4)
|
|
|
|
| 60 |
|
| 61 |
# ββ Load model (cached after first download) ββββββββββββββββββββββ
|
| 62 |
def load_model():
|
| 63 |
+
ckpt_path = hf_hub_download(repo_id="S-4-G-4-R/brain-tumor-efficientnet-b3", filename=CKPT_FILE)
|
| 64 |
ckpt = torch.load(ckpt_path, map_location=DEVICE, weights_only=False)
|
| 65 |
|
| 66 |
n_classes = ckpt.get("num_classes", 4)
|