yamildiego commited on
Commit
bfbd999
·
1 Parent(s): 868133d
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -50,10 +50,11 @@ class EndpointHandler():
50
  # providers=["CPUExecutionProvider"],
51
  # )
52
 
53
- dir_path = os.path.join("/repository", "models", "antelopev2")
54
  print(dir_path)
 
55
 
56
- self.app = FaceAnalysis(name="antelopev2", root="/repository", providers=["CPUExecutionProvider"])
57
  self.app.prepare(ctx_id=0, det_size=(640, 640))
58
 
59
  openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
 
50
  # providers=["CPUExecutionProvider"],
51
  # )
52
 
53
+ dir_path = os.path.join("", "models", "antelopev2")
54
  print(dir_path)
55
+ print(model_dir)
56
 
57
+ self.app = FaceAnalysis(name="antelopev2", providers=["CPUExecutionProvider"])
58
  self.app.prepare(ctx_id=0, det_size=(640, 640))
59
 
60
  openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")