Update handler.py
Browse files- handler.py +2 -1
handler.py
CHANGED
|
@@ -5,7 +5,8 @@ class EndpointHandler:
|
|
| 5 |
"""
|
| 6 |
Initialize the EndpointHandler with the YOLO model from the given directory.
|
| 7 |
"""
|
| 8 |
-
|
|
|
|
| 9 |
|
| 10 |
def preprocess(self, inputs):
|
| 11 |
"""
|
|
|
|
| 5 |
"""
|
| 6 |
Initialize the EndpointHandler with the YOLO model from the given directory.
|
| 7 |
"""
|
| 8 |
+
# Update the model file name to match your actual file
|
| 9 |
+
self.model = YOLO(f"{model_dir}/tr_roi_finetune_130_large.pt")
|
| 10 |
|
| 11 |
def preprocess(self, inputs):
|
| 12 |
"""
|