emreyoleridev commited on
Commit
c00a9c7
·
verified ·
1 Parent(s): 5b12bba

Upload app/inference/predictor.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app/inference/predictor.py +1 -1
app/inference/predictor.py CHANGED
@@ -40,7 +40,7 @@ class BrainTumorPredictor:
40
  try:
41
  from huggingface_hub import hf_hub_download
42
  # Kendi Hugging Face kullanıcı adınızı aşağıya yazın (emreyoleridev kısmına)
43
- hf_repo_id = "emreyoleridev/medvision-unet"
44
  model_path = hf_hub_download(repo_id=hf_repo_id, filename="best_unet_model.pth")
45
  print(f"[INFO] Successfully downloaded weights to {model_path}")
46
  except Exception as e:
 
40
  try:
41
  from huggingface_hub import hf_hub_download
42
  # Kendi Hugging Face kullanıcı adınızı aşağıya yazın (emreyoleridev kısmına)
43
+ hf_repo_id = "emreyoleridev/brain-neural-bound-tumor-model"
44
  model_path = hf_hub_download(repo_id=hf_repo_id, filename="best_unet_model.pth")
45
  print(f"[INFO] Successfully downloaded weights to {model_path}")
46
  except Exception as e: