oscarfu0501 commited on
Commit
c0d268d
1 Parent(s): bb2a943

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +6 -2
handler.py CHANGED
@@ -20,12 +20,16 @@ class EndpointHandler:
20
  self.model = YOLO("./best.pt")
21
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
22
  inputs = data.get("inputs")
 
 
23
  isurl = inputs.get("isurl")
 
 
24
  path = inputs.get("path")
 
25
  print(path)
26
  # path = "http://10.10.2.100/cam-lo.jpg"
27
 
28
- model = self.model
29
  ########################### Load Image #################################
30
  if(isurl): # for url set isurl = 1
31
  print("checkpoint 2-1")
@@ -49,7 +53,7 @@ class EndpointHandler:
49
  # t-shirts-detector/1
50
  # mainmodel/2
51
  #result = self.CLIENT.infer(path, model_id="mainmodel/2")
52
- result = model(img)
53
  #annotated_frame = result[0].plot()
54
  detections = result[0].boxes
55
  #print(result[0].boxes.xyxy)
 
20
  self.model = YOLO("./best.pt")
21
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
22
  inputs = data.get("inputs")
23
+ print("in call")
24
+
25
  isurl = inputs.get("isurl")
26
+ print("in isurl")
27
+
28
  path = inputs.get("path")
29
+ print("is path")
30
  print(path)
31
  # path = "http://10.10.2.100/cam-lo.jpg"
32
 
 
33
  ########################### Load Image #################################
34
  if(isurl): # for url set isurl = 1
35
  print("checkpoint 2-1")
 
53
  # t-shirts-detector/1
54
  # mainmodel/2
55
  #result = self.CLIENT.infer(path, model_id="mainmodel/2")
56
+ result = self.model(img)
57
  #annotated_frame = result[0].plot()
58
  detections = result[0].boxes
59
  #print(result[0].boxes.xyxy)