Upload app/inference/predictor.py with huggingface_hub
Browse files
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/
|
| 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:
|