Update model.py

model.py

@@ -1,6 +1,7 @@
 import torch
 import torch.nn as nn
 import numpy as np
+import os
 from torchvision import transforms
 from torchvision.models import resnet18, ResNet18_Weights
 
@@ -22,7 +23,7 @@ class ResNet18_M3(nn.Module):
 
     def get_instance(self):
         DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        weights_path = "./resnet18_m3_best.pth"
+        weights_path = os.path.join(os.path.dirname(__file__), "resnet18_m3_best.pth")
     
         model = ResNet18_M3().to(DEVICE)