banao-tech commited on
Commit
4b6cfea
·
verified ·
1 Parent(s): f36b296

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -25,7 +25,7 @@ class ModelManager:
25
  self.model = None
26
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
27
 
28
- def load_models(self):
29
  """Initialize all required models"""
30
  try:
31
  # Load YOLO model
@@ -212,4 +212,4 @@ async def process_image(
212
 
213
  if __name__ == "__main__":
214
  import uvicorn
215
- uvicorn.run(app, host="0.0.0.0", port=8000)
 
25
  self.model = None
26
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
27
 
28
+ def load_models(self) -> bool:
29
  """Initialize all required models"""
30
  try:
31
  # Load YOLO model
 
212
 
213
  if __name__ == "__main__":
214
  import uvicorn
215
+ uvicorn.run(app, host="0.0.0.0", port=8000)