Update app.py

app.py

@@ -8,8 +8,8 @@ import numpy as np
 from typing import Optional, Tuple, List
 
 # Initialize CUDA
-zero = torch.Tensor([0]).cuda()
-print(f"Initial device: {zero.device}")
+#zero = torch.Tensor([0]).cuda()
+#print(f"Initial device: {zero.device}")
 
 # Model configurations
 BASE_MODEL = "Qwen/Qwen-Image-Edit-2509"  # Qwen Image Edit model
@@ -41,7 +41,7 @@ def face_swap(
     """
     Perform head swap using Qwen-Image-Edit with multiple LoRAs
     """
-    print(f"GPU device: {zero.device}")
+    #print(f"GPU device: {zero.device}")
     
     # Validate inputs
     if body_image is None or face_image is None: