Rahaf2001 commited on
Commit
d051cb0
·
verified ·
1 Parent(s): 60a59f0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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(" Model ready")
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")