Update app.py

app.py

@@ -65,10 +65,25 @@ async def load_model():
     t0 = time.time()
 
     import torch
-    model = torch.load(MODEL_PATH, map_location=DEVICE, weights_only=False)
-    model.eval()
-
-    print(f"✅ Model loaded in {time.time()-t0:.2f}s")
+    checkpoint = torch.load(MODEL_PATH, map_location=DEVICE, weights_only=False)
+
+    # Checkpoint se model nikalo
+    if isinstance(checkpoint, dict):
+        if 'model' in checkpoint:
+            model = checkpoint['model']
+        elif 'state_dict' in checkpoint:
+            model = checkpoint['state_dict']
+        else:
+            # Sirf weights hain — keys print karo debug ke liye
+            print(f"Checkpoint keys: {list(checkpoint.keys())}")
+            model = checkpoint
+    else:
+        model = checkpoint
+    
+    if hasattr(model, 'eval'):
+        model.eval()
+    
+        print(f"✅ Model loaded in {time.time()-t0:.2f}s")
 
 
 @app.get("/")