Update app.py

app.py

@@ -1,7 +1,13 @@
-from my_model import MyModel  # import your architecture
+import torch
 
-model = MyModel(...)  # initialize with same architecture
-state_dict = torch.load("model/model.pt", map_location=device)  # works with weights_only=True
-model.load_state_dict(state_dict)
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+MODEL_PATH = "model/model.pt"
+
+# load full model (from your .pt file)
+print(f"Loading model from: {MODEL_PATH}")
+model = torch.load(MODEL_PATH, map_location=device, weights_only=False)
 model.to(device)
 model.eval()
+
+print("Model loaded successfully!")