collinbarnwell commited on
Commit
0e8116a
·
verified ·
1 Parent(s): 2cef658

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -31,7 +31,7 @@ class EndpointHandler:
31
  print("-----------------------------------")
32
  print(f"PATH: {path}")
33
  print("-----------------------------------")
34
- self._pipeline = Pipeline.from_pretrained(path)
35
  HYPER_PARAMETERS = {
36
  "segmentation": {
37
  "min_duration_off": 3.0,
 
31
  print("-----------------------------------")
32
  print(f"PATH: {path}")
33
  print("-----------------------------------")
34
+ self._pipeline = Pipeline.from_pretrained(path.strip("/"))
35
  HYPER_PARAMETERS = {
36
  "segmentation": {
37
  "min_duration_off": 3.0,