Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from ultralytics import YOLO
|
|
| 6 |
import time
|
| 7 |
|
| 8 |
app = Flask(__name__)
|
| 9 |
-
model = YOLO("
|
| 10 |
|
| 11 |
KNOWN_WIDTHS = {
|
| 12 |
"person": 50, "chair": 40, "bottle": 7,
|
|
@@ -74,5 +74,4 @@ def detect():
|
|
| 74 |
return jsonify({"detections": detections})
|
| 75 |
|
| 76 |
if __name__ == "__main__":
|
| 77 |
-
app.run(host="0.0.0.0", port=
|
| 78 |
-
|
|
|
|
| 6 |
import time
|
| 7 |
|
| 8 |
app = Flask(__name__)
|
| 9 |
+
model = YOLO("yolov8s.pt")
|
| 10 |
|
| 11 |
KNOWN_WIDTHS = {
|
| 12 |
"person": 50, "chair": 40, "bottle": 7,
|
|
|
|
| 74 |
return jsonify({"detections": detections})
|
| 75 |
|
| 76 |
if __name__ == "__main__":
|
| 77 |
+
app.run(host="0.0.0.0", port=5000)
|
|
|