Hugging Face's logo

MakeNTU-v2024
/
Image-preprocessor

Model card Files Community
oscarfu0501 commited on
Commit
227439e
·
verified ·
1 Parent(s): d132e28

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -15,7 +15,7 @@ class EndpointHandler:
15
  isurl = inputs.get("isurl")
16
  path = inputs.get("path")
17
 
18
- model = YOLO("best.pt")
19
  ########################### Load Image #################################
20
  if(isurl): # for url set isurl = 1
21
  print("checkpoint 2-1")
 
15
  isurl = inputs.get("isurl")
16
  path = inputs.get("path")
17
 
18
+ model = YOLO("Image-preprocessor/best.pt")
19
  ########################### Load Image #################################
20
  if(isurl): # for url set isurl = 1
21
  print("checkpoint 2-1")