X1ng1 commited on
Commit
6fde022
·
1 Parent(s): b7e3b0a

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
- import uvicorn
333
- uvicorn.run(app, host="0.0.0.0", port=8000)
 
 
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 self.base_model_path.exists():
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