Upload 4 files

README.md

@@ -0,0 +1,32 @@
+---
+title: Video FaceSwap GPU Only
+emoji: 🎭
+colorFrom: blue
+colorTo: purple
+sdk: docker
+app_port: 7860
+suggested_hardware: a10g-small
+startup_duration_timeout: 1h
+pinned: false
+---
+
+# Video FaceSwap GPU Only
+
+This Space is strict GPU-only. It refuses to process video unless ONNX Runtime exposes CUDAExecutionProvider.
+
+## Environment variables
+
+- `MODEL_REPO_ID`: default `ezioruan/inswapper_128.onnx`
+- `MODEL_FILENAME`: default `inswapper_128.onnx`
+- `MODEL_REVISION`: optional branch, tag, or commit hash
+- `HF_TOKEN`: recommended for gated/private repos and better Hub rate limits
+- `FACE_MODEL_NAME`: default `buffalo_l`
+- `DETECTION_SIZE`: default `640`
+- `MAX_FRAMES`: optional frame cap for testing, default `0`
+
+## Deploy
+
+1. Create a Docker Space.
+2. Select a paid GPU hardware tier such as A10G small.
+3. Add these files.
+4. Add `HF_TOKEN` if needed.

app.py

@@ -40,14 +40,15 @@ def available_providers():
 
 
 def gpu_ready():
-    return "CUDAExecutionProvider" in available_providers()
+    providers = available_providers()
+    return "CUDAExecutionProvider" in providers, providers
 
 
 def gpu_guard():
-    providers = available_providers()
-    if "CUDAExecutionProvider" not in providers:
+    ok, providers = gpu_ready()
+    if not ok:
         raise RuntimeError(
-            "GPU-only mode: CUDAExecutionProvider is missing. "
+            "GPU device not attached. CUDAExecutionProvider is missing. "
             f"Available providers: {providers}"
         )
 
@@ -112,8 +113,8 @@ def swap_frame(frame_bgr, source_face, analyzer, swapper):
 
 
 def startup_status():
-    providers = available_providers()
-    return f"Available providers: {providers} | GPU ready: {gpu_ready()}"
+    ok, providers = gpu_ready()
+    return f"GPU ready: {ok} | Providers: {providers}"
 
 
 def process_video(face_image_path, video_path, progress=gr.Progress(track_tqdm=False)):
@@ -163,7 +164,7 @@ def process_video(face_image_path, video_path, progress=gr.Progress(track_tqdm=F
 
 with gr.Blocks(title="Video FaceSwap GPU Only") as demo:
     gr.Markdown("""# Video FaceSwap GPU Only
-This Space will only run when ONNX Runtime exposes CUDAExecutionProvider. If not, it refuses to process video.""")
+This Space only works when a real GPU is attached. If CUDA is missing, processing is blocked.""")
     gr.Textbox(label="GPU status", value=startup_status(), interactive=False)
     with gr.Row():
         face_image = gr.Image(type="filepath", label="Source face image")