Update models.py

models.py

@@ -70,7 +70,7 @@ def load_face_analysis():
 
     try:
         antelope_download = snapshot_download(repo_id="DIAMONIK7777/antelopev2", local_dir="/data/models/antelopev2")
-        face_app = FaceAnalysis(name='antelopev2', root='/data', providers=['CUDAExecutionProvider'])
+        face_app = FaceAnalysis(name='antelopev2', root='/data', providers=['CPUExecutionProvider']) # Changed from CUDA to CPU
         face_app.prepare(ctx_id=0, det_size=(640, 640))
         return face_app, True
         
@@ -92,7 +92,7 @@ def load_depth_detector():
     try:
         print("  Attempting LeresDetector (highest quality)...")
         leres_depth = LeresDetector.from_pretrained("lllyasviel/Annotators")
-        leres_depth.to(device)
+        # leres_depth.to(device)
         print("  [OK] LeresDetector loaded successfully")
         return leres_depth, 'leres', True
     except Exception as e:
@@ -102,7 +102,7 @@ def load_depth_detector():
     try:
         print("  Attempting ZoeDetector (fallback #1)...")
         zoe_depth = ZoeDetector.from_pretrained("lllyasviel/Annotators")
-        zoe_depth.to(device)
+        # zoe_depth.to(device)
         print("  [OK] ZoeDetector loaded successfully")
         return zoe_depth, 'zoe', True
     except Exception as e:
@@ -112,7 +112,7 @@ def load_depth_detector():
     try:
         print("  Attempting MidasDetector (fallback #2)...")
         midas_depth = MidasDetector.from_pretrained("lllyasviel/Annotators")
-        midas_depth.to(device)
+        # midas_depth.to(device)
         print("  [OK] MidasDetector loaded successfully")
         return midas_depth, 'midas', True
     except Exception as e:
@@ -127,7 +127,7 @@ def load_openpose_detector():
     print("Loading OpenPose detector...")
     try:
         openpose = OpenposeDetector.from_pretrained("lllyasviel/Annotators")
-        openpose.to(device)
+        # openpose.to(device)
         print("  [OK] OpenPose loaded successfully")
         return openpose, True
     except Exception as e:
@@ -154,7 +154,7 @@ def load_controlnets():
     controlnet_depth = ControlNetModel.from_pretrained(
         "xinsir/controlnet-depth-sdxl-1.0",
         torch_dtype=dtype
-    ).to(device)
+    )#.to(device)
     print("  [OK] ControlNet Depth loaded")
 
     # --- NEW: Load OpenPose ControlNet ---
@@ -163,7 +163,7 @@ def load_controlnets():
         controlnet_openpose = ControlNetModel.from_pretrained(
             "xinsir/controlnet-openpose-sdxl-1.0",
             torch_dtype=dtype
-        ).to(device)
+        )#.to(device)
         print("  [OK] ControlNet OpenPose loaded")
     except Exception as e:
         print(f"  [WARNING] ControlNet OpenPose not available: {e}")
@@ -176,7 +176,7 @@ def load_controlnets():
             "InstantX/InstantID",
             subfolder="ControlNetModel",
             torch_dtype=dtype
-        ).to(device)
+        )#.to(device)
         print("  [OK] InstantID ControlNet loaded successfully")
         # Return all three models
         return controlnet_depth, controlnet_instantid, controlnet_openpose, True
@@ -194,7 +194,7 @@ def load_image_encoder():
             "h94/IP-Adapter",
             subfolder="models/image_encoder",
             torch_dtype=dtype
-        ).to(device)
+        )#.to(device)
         print("  [OK] CLIP Image Encoder loaded successfully")
         return image_encoder
     except Exception as e:
@@ -436,7 +436,7 @@ def load_caption_model():
         caption_model = AutoModelForCausalLM.from_pretrained(
             "microsoft/git-large-coco",
             torch_dtype=dtype
-        ).to(device)
+        )#.to(device)
         print("  [OK] GIT-Large model loaded (produces detailed captions)")
         return caption_processor, caption_model, True, 'git'
     except Exception as e1:
@@ -451,7 +451,7 @@ def load_caption_model():
             caption_model = BlipForConditionalGeneration.from_pretrained(
                 "Salesforce/blip-image-captioning-base",
                 torch_dtype=dtype
-            ).to(device)
+            )#.to(device)
             print("  [OK] BLIP base model loaded (standard captions)")
             return caption_processor, caption_model, True, 'blip'
         except Exception as e2: