Update app.py

app.py

@@ -1,13 +1,9 @@
 import torch
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
 MODEL_PATH = "model/model.pt"
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# 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!")