Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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)
|