fasd

handler.py

@@ -3,11 +3,13 @@ import torch
 import numpy as np
 import torch.nn.functional as F
 from serkan import SimpleUpscaleModel
+import os
 class EndpointHandler():
-    def __init__(self, path="model_weights.pth"):
+    def __init__(self, path="."):
         # load the optimized model
         self.model = SimpleUpscaleModel()
-        self.model.load_state_dict(torch.load("model_weights.pth"))
+        model_path = os.path.join(path, "model_weights.pth")
+        self.model.load_state_dict(torch.load(model_path))