Update app.py

app.py

@@ -1,10 +1,10 @@
+import os
 import spaces
 import torch
 from diffusers import QwenImageEditPlusPipeline
 from PIL import Image
 import gradio as gr
 
-
 HF_BASE_MODEL = "Qwen/Qwen-Image-Edit-2509"
 BFS_LORA = "Alissonerdx/BFS-Best-Face-Swap"
 BFS_LORA_WEIGHT = "bfs_head_v3_qwen_image_edit_2509.safetensors"  # Head V3 (recommended)
@@ -14,13 +14,16 @@ BFS_LORA_WEIGHT = "bfs_head_v3_qwen_image_edit_2509.safetensors"  # Head V3 (rec
 # Lưu device để dùng lại trong quá trình suy luận.
 EXEC_DEVICE = "cpu"
 
+# Cho phép ép dùng CPU nếu GPU yếu hoặc hay abort (ZeroGPU không ổn định).
+FORCE_CPU = bool(int(os.getenv("FORCE_CPU", "0")))
+
 
 @spaces.GPU  # bắt buộc cho ZeroGPU
 def load_pipeline():
     global EXEC_DEVICE
 
-    # Ưu tiên GPU của ZeroGPU; fallback CPU khi chạy local.
-    device = "cuda" if torch.cuda.is_available() else "cpu"
+    # Ưu tiên GPU của ZeroGPU; cho phép ép CPU để tránh abort/OOM.
+    device = "cuda" if torch.cuda.is_available() and not FORCE_CPU else "cpu"
     dtype = torch.float16 if device == "cuda" else torch.float32
     EXEC_DEVICE = device