S-4-G-4-R commited on
Commit
b4f632d
Β·
verified Β·
1 Parent(s): 47047a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=HF_REPO_ID, filename=CKPT_FILE)
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)