Update generator.py

generator.py

@@ -145,8 +145,7 @@ class RetroArtConverter:
             print(f"[INFO] Verification skipped: {e}")
         print("============================\n")
     
-
-        def get_depth_map(self, image):
+    def get_depth_map(self, image):
             """Generate depth map using Zoe Depth"""
             if self.zoe_depth is not None:
                 try:
@@ -619,7 +618,7 @@ class RetroArtConverter:
                     # Reshape for Resampler: [1, 1, 512]
                     face_emb_tensor = face_emb_tensor.reshape(1, -1, 512)
                     
-                    # Pass through Resampler: [1, 1, 512] → [1, 16, 2048]
+                    # Pass through Resampler: [1, 1, 512] → [1, 16, 2048]
                     face_proj_embeds = self.image_proj_model(face_emb_tensor)
                     
                     # Scale with identity preservation
@@ -707,4 +706,4 @@ class RetroArtConverter:
         return generated_image
 
 
-print("[OK] Generator class ready")
+print("[OK] Generator class ready")