Update handler.py

handler.py

@@ -8,20 +8,20 @@ import os
 import gdown
 from PIL import Image
 import io
-import http.client
+# import http.client
 
-http.client.HTTPConnection._http_vsn = 10
-http.client.HTTPConnection._http_vsn_str = 'HTTP/1.0'
+# http.client.HTTPConnection._http_vsn = 10
+# http.client.HTTPConnection._http_vsn_str = 'HTTP/1.0'
 
 class EndpointHandler:
     def __init__(self, path='.'):  # pass api key to model
         # current_directory = os.getcwd()
         # print("Current working directory:", current_directory)
         
-        url = "https://drive.google.com/file/d/1jB8sDYYOTfuF7B1PMcDjkm5R7huv97Wm"
-        gdown.download(url, 'best.pt', quiet=False)
+        url = "https://drive.google.com/file/d/1jB8sDYYOTfuF7B1PMcDjkm5R7huv97Wm/view?usp=sharing"
+        gdown.download(url, './best.pt', quiet=False)
         
-        self.model = YOLO("best.pt")
+        self.model = YOLO("./best.pt")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         inputs = data.get("inputs")
         print("in call")
@@ -32,7 +32,7 @@ class EndpointHandler:
         path = inputs.get("path")
         print("is path")
         print(path)
-        # path = "http://10.10.2.100/cam-lo.jpg"
+        path = "http://10.10.2.100/cam-lo.jpg"
         model = self.model
 ###########################  Load Image  #################################       
         if(isurl): # for url set isurl = 1