Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -6,7 +6,7 @@ import io, requests, os, tempfile
|
|
| 6 |
import cv2
|
| 7 |
|
| 8 |
app = FastAPI()
|
| 9 |
-
|
| 10 |
app.add_middleware(
|
| 11 |
CORSMiddleware,
|
| 12 |
allow_origins=["*"],
|
|
@@ -22,7 +22,7 @@ def download_model():
|
|
| 22 |
r = requests.get(MODEL_URL)
|
| 23 |
with open("best.pt", "wb") as f:
|
| 24 |
f.write(r.content)
|
| 25 |
-
print("
|
| 26 |
|
| 27 |
download_model()
|
| 28 |
model = YOLO("best.pt")
|
|
|
|
| 6 |
import cv2
|
| 7 |
|
| 8 |
app = FastAPI()
|
| 9 |
+
#
|
| 10 |
app.add_middleware(
|
| 11 |
CORSMiddleware,
|
| 12 |
allow_origins=["*"],
|
|
|
|
| 22 |
r = requests.get(MODEL_URL)
|
| 23 |
with open("best.pt", "wb") as f:
|
| 24 |
f.write(r.content)
|
| 25 |
+
print(" Model ready")
|
| 26 |
|
| 27 |
download_model()
|
| 28 |
model = YOLO("best.pt")
|