Spaces:
Running
Running
fixed kiuyha bubble detector downloads
Browse files
__pycache__/db.cpython-312.pyc
ADDED
|
Binary file (13.3 kB). View file
|
|
|
__pycache__/helpers.cpython-312.pyc
ADDED
|
Binary file (2.65 kB). View file
|
|
|
main.py
CHANGED
|
@@ -329,5 +329,6 @@ def test(img_path: Optional[str] = None):
|
|
| 329 |
print(f"{img_path} does not exist")
|
| 330 |
|
| 331 |
if __name__ == "__main__":
|
| 332 |
-
|
| 333 |
-
|
|
|
|
|
|
| 329 |
print(f"{img_path} does not exist")
|
| 330 |
|
| 331 |
if __name__ == "__main__":
|
| 332 |
+
main()
|
| 333 |
+
# import uvicorn
|
| 334 |
+
# uvicorn.run(app, host="0.0.0.0", port=8000)
|
services/bubble_detector_kiuyha_service.py
CHANGED
|
@@ -15,7 +15,7 @@ class Bubble_Detector_Kiuyha_Service:
|
|
| 15 |
else:
|
| 16 |
model_path = Path(model_path)
|
| 17 |
|
| 18 |
-
if not
|
| 19 |
print(f"Kiuyha model not found at {model_path}. Attempting to download")
|
| 20 |
self.load_model()
|
| 21 |
|
|
@@ -73,7 +73,7 @@ class Bubble_Detector_Kiuyha_Service:
|
|
| 73 |
|
| 74 |
downloaded_path = hf_hub_download(
|
| 75 |
repo_id="Kiuyha/Manga-Bubble-YOLO",
|
| 76 |
-
filename="model.pt",
|
| 77 |
local_dir=self.base_model_path
|
| 78 |
)
|
| 79 |
|
|
|
|
| 15 |
else:
|
| 16 |
model_path = Path(model_path)
|
| 17 |
|
| 18 |
+
if not model_path.exists():
|
| 19 |
print(f"Kiuyha model not found at {model_path}. Attempting to download")
|
| 20 |
self.load_model()
|
| 21 |
|
|
|
|
| 73 |
|
| 74 |
downloaded_path = hf_hub_download(
|
| 75 |
repo_id="Kiuyha/Manga-Bubble-YOLO",
|
| 76 |
+
filename="weights/yolo26n.pt", #"model.pt",
|
| 77 |
local_dir=self.base_model_path
|
| 78 |
)
|
| 79 |
|