fixed path

app.py

@@ -14,7 +14,7 @@ import inference as inf
 def colorize_image(image):
      # Load the model
     # file_path = 'ImageColorizationModel10.pth'
-    file_path = r'model\model_final.pth'
+    file_path = './model/model_final.pth'
     model_2 = inf.load_model(model_class=MainModel, file_path=file_path)
     output_img = inf.predict_color(model_2, image=image)
     return output_img

inference.py

@@ -98,7 +98,7 @@ def load_model(model_class, file_path):
     model.load_state_dict(torch.load(file_path, map_location=device))
 
     resnet_weights = torch.load(file_path)
-    resnet_weights = torch.load(r"model\res18-unet.pt")
+    resnet_weights = torch.load("./model/res18-unet.pt")
     resnet_state_dict = resnet_weights['state_dict'] if 'state_dict' in resnet_weights else resnet_weights
 
     model_dict = model.state_dict()