Update app.py

app.py

@@ -29,14 +29,22 @@ class RetroArtConverter:
         self.device = device
         self.dtype = dtype
         
-        # Initialize face analysis for InstantID
+        # Initialize face analysis for InstantID (optional)
         print("Loading face analysis model...")
-        self.face_app = FaceAnalysis(
-            name='antelopev2',
-            root='./models/insightface',
-            providers=['CUDAExecutionProvider', 'CPUExecutionProvider']
-        )
-        self.face_app.prepare(ctx_id=0, det_size=(640, 640))
+        try:
+            self.face_app = FaceAnalysis(
+                name='antelopev2',
+                root='./models/insightface',
+                providers=['CUDAExecutionProvider', 'CPUExecutionProvider']
+            )
+            self.face_app.prepare(ctx_id=0, det_size=(640, 640))
+            print("✓ Face analysis model loaded successfully")
+            self.face_detection_enabled = True
+        except Exception as e:
+            print(f"⚠️ Face detection not available: {e}")
+            print("Continuing without face detection (will still work fine)")
+            self.face_app = None
+            self.face_detection_enabled = False
         
         # Load ControlNet for depth
         print("Loading ControlNet depth model...")
@@ -151,9 +159,16 @@ class RetroArtConverter:
     
     def detect_faces(self, image):
         """Detect faces in the image using antelopev2"""
-        img_array = np.array(image)
-        faces = self.face_app.get(img_array)
-        return faces
+        if not self.face_detection_enabled or self.face_app is None:
+            return []
+        
+        try:
+            img_array = np.array(image)
+            faces = self.face_app.get(img_array)
+            return faces
+        except Exception as e:
+            print(f"Face detection error: {e}")
+            return []
     
     def calculate_target_size(self, original_width, original_height, max_dimension=1024):
         """Calculate target size maintaining aspect ratio"""
@@ -350,4 +365,4 @@ if __name__ == "__main__":
         server_port=7860,
         share=False,
         show_api=True  # Enable API
-    )
+    )