MakeNTU-v2024
/

Image-preprocessor

Model card Files Files and versions

oscarfu0501 commited on May 4, 2024

Commit

d132e28

·

verified ·

1 Parent(s): 01e9ad4

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -15,7 +15,7 @@ class EndpointHandler:
         isurl = inputs.get("isurl")
         path = inputs.get("path")
-        model = YOLO("./best.pt")
 ###########################  Load Image  #################################
         if(isurl): # for url set isurl = 1
             print("checkpoint 2-1")

         isurl = inputs.get("isurl")
         path = inputs.get("path")
+        model = YOLO("best.pt")
 ###########################  Load Image  #################################
         if(isurl): # for url set isurl = 1
             print("checkpoint 2-1")