MakeNTU-v2024
/

Image-preprocessor

Inference Endpoints

Model card Files Files and versions Community

oscarfu0501 commited on May 4

Commit

452921c

•

1 Parent(s): 227439e

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("Image-preprocessor/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("MakeNTU-v2024/Image-preprocessor/best.pt")
 ###########################  Load Image  #################################
         if(isurl): # for url set isurl = 1
             print("checkpoint 2-1")