Update handler.py
Browse files- handler.py +4 -4
handler.py
CHANGED
@@ -15,10 +15,10 @@ class EndpointHandler():
|
|
15 |
|
16 |
# Check different possible locations
|
17 |
possible_paths = [
|
18 |
-
os.path.join(base_path, "checkpoint-
|
19 |
-
os.path.join(".", "checkpoint-
|
20 |
-
os.path.join("/repository", "checkpoint-
|
21 |
-
"checkpoint-
|
22 |
]
|
23 |
|
24 |
for check_path in possible_paths:
|
|
|
15 |
|
16 |
# Check different possible locations
|
17 |
possible_paths = [
|
18 |
+
os.path.join(base_path, "checkpoint-36571"),
|
19 |
+
os.path.join(".", "checkpoint-36571"),
|
20 |
+
os.path.join("/repository", "checkpoint-36571"),
|
21 |
+
"checkpoint-36571"
|
22 |
]
|
23 |
|
24 |
for check_path in possible_paths:
|