Update handler.py
Browse files- 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,
|