added more debugging for model load

app.py

@@ -45,7 +45,12 @@ def load_model():
         checkpoint = torch.load(model_path, map_location=DEVICE)
         
         gen_model = Generator(z_dim=Z_DIM).to(DEVICE)
-        gen_model.load_state_dict(checkpoint["gen_state_dict"], strict=False)
+        missing, unexpected = gen_model.load_state_dict(
+            checkpoint["gen_state_dict"], strict=False
+        )
+
+        print("Unexpected keys: ", unexpected)
+        print("Missing keys: ", missing)
         gen_model.eval()
         print("SUCCESS: Petrol Pump GAN is live!")
     except Exception as e:
@@ -87,10 +92,7 @@ def generate_random():
     """Endpoint 1: Purely random generation for 'Discovery'."""
     if gen_model is None: raise HTTPException(status_code=503)
     
-    with torch.inference_mode():
-        seed = torch.seed() 
-        torch.manual_seed(seed)
-        
+    with torch.inference_mode():        
         noise = torch.randn(1, Z_DIM, device=DEVICE)
         print("NOISE:", noise[0, :5])
         fake_img = gen_model(noise)