Update handler.py
Browse filesmove to -04 model which is made with Con's sort
- handler.py +1 -1
handler.py
CHANGED
|
@@ -8,7 +8,7 @@ class EndpointHandler():
|
|
| 8 |
def __init__(self, path=""):
|
| 9 |
# Load model and weights
|
| 10 |
self.model = get_model()
|
| 11 |
-
weights_path = os.path.join(path, "doge_223_sd-
|
| 12 |
self.model.load_state_dict(torch.load(weights_path, map_location="cpu"))
|
| 13 |
self.model.eval()
|
| 14 |
|
|
|
|
| 8 |
def __init__(self, path=""):
|
| 9 |
# Load model and weights
|
| 10 |
self.model = get_model()
|
| 11 |
+
weights_path = os.path.join(path, "doge_223_sd-04.bin")
|
| 12 |
self.model.load_state_dict(torch.load(weights_path, map_location="cpu"))
|
| 13 |
self.model.eval()
|
| 14 |
|