Update model

model

@@ -1,5 +1,7 @@
-mkdir -p model
-cp /path/to/your/model.pt model/model.pt
-git add model/model.pt
-git commit -m "Add model weights"
-git push
+from my_model import MyModel  # import your model class
+
+model = MyModel(...)  # init with same architecture
+state_dict = torch.load("model/model.pt", map_location=device)
+model.load_state_dict(state_dict)
+model.to(device)
+model.eval()