Update inference.py

inference.py

@@ -1,7 +1,7 @@
 import torch
 from model import *  # import extracted model classes
 
-def load_model(weights_path="model1.pt", device=None):
+def load_model(weights_path="model.pt", device=None):
     if device is None:
         device = "cuda" if torch.cuda.is_available() else "cpu"
     # Replace 'YourModelClass' with the actual class name from model.py