Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,11 +8,6 @@ import spaces
|
|
| 8 |
|
| 9 |
@spaces.GPU
|
| 10 |
|
| 11 |
-
import cv2
|
| 12 |
-
import torch
|
| 13 |
-
import os
|
| 14 |
-
from ultralytics import YOLO
|
| 15 |
-
|
| 16 |
class CrowdDetection:
|
| 17 |
def __init__(self, model_path="yolov8n.pt"):
|
| 18 |
"""Initialize the YOLO model once to avoid PicklingError."""
|
|
@@ -225,5 +220,5 @@ interface = gr.Interface(
|
|
| 225 |
)
|
| 226 |
|
| 227 |
if __name__ == "__main__":
|
| 228 |
-
interface.launch()
|
| 229 |
|
|
|
|
| 8 |
|
| 9 |
@spaces.GPU
|
| 10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
class CrowdDetection:
|
| 12 |
def __init__(self, model_path="yolov8n.pt"):
|
| 13 |
"""Initialize the YOLO model once to avoid PicklingError."""
|
|
|
|
| 220 |
)
|
| 221 |
|
| 222 |
if __name__ == "__main__":
|
| 223 |
+
interface.launch(debug=True)
|
| 224 |
|