Update handler.py
Browse files- handler.py +3 -0
handler.py
CHANGED
|
@@ -28,6 +28,9 @@ import torch
|
|
| 28 |
class EndpointHandler:
|
| 29 |
def __init__(self, path=""):
|
| 30 |
# initialize pretrained pipeline
|
|
|
|
|
|
|
|
|
|
| 31 |
self._pipeline = Pipeline.from_pretrained(path)
|
| 32 |
HYPER_PARAMETERS = {
|
| 33 |
"segmentation": {
|
|
|
|
| 28 |
class EndpointHandler:
|
| 29 |
def __init__(self, path=""):
|
| 30 |
# initialize pretrained pipeline
|
| 31 |
+
print("-----------------------------------")
|
| 32 |
+
print(f"PATH: {path}")
|
| 33 |
+
print("-----------------------------------")
|
| 34 |
self._pipeline = Pipeline.from_pretrained(path)
|
| 35 |
HYPER_PARAMETERS = {
|
| 36 |
"segmentation": {
|