Add ZeroGPU decorator + ffmpeg package + py3.10 pin

README.md

@@ -1,17 +1,5 @@
 ---
 title: Qwen Image Edit Rapid Aio
-emoji: 🚀
-colorFrom: indigo
-colorTo: blue
 sdk: gradio
-sdk_version: 5.0.0
-app_file: app.py
-pinned: false
+python_version: "3.10"
 ---
-
-# Qwen Image Edit Rapid Aio
-
-Space initialisé automatiquement pour être fonctionnel.
-
-- Hardware demandé: **zero-a10g** (quota HF ZeroGPU)
-- Ce commit initialise l'app Gradio et la phase de build

app.py

@@ -3,9 +3,14 @@ from datetime import datetime
 import gradio as gr
 from PIL import Image, ImageDraw
 
+try:
+    import spaces
+except Exception:
+    spaces = None
+
 TITLE = "Qwen Image Edit Rapid Aio"
 MODEL_HINT = "qwen-image-edit-rapid-aio"
-MODE = "image"  # image | video
+MODE = "image"
 
 
 def _render_image(prompt: str) -> Image.Image:
@@ -30,7 +35,6 @@ def _render_image(prompt: str) -> Image.Image:
 def _render_video(prompt: str) -> str:
     import imageio.v2 as imageio
     import numpy as np
-
     out_path = "/tmp/out.mp4"
     frames = []
     w, h = 640, 360
@@ -44,19 +48,25 @@ def _render_video(prompt: str) -> str:
     return out_path
 
 
-def infer(prompt: str):
+def _infer(prompt: str):
     if MODE == "video":
         return _render_video(prompt), f"✅ {TITLE} OK (CPU-safe)."
     return _render_image(prompt), f"✅ {TITLE} OK (CPU-safe)."
 
 
+if spaces is not None:
+    infer = spaces.GPU(duration=60)(_infer)
+else:
+    infer = _infer
+
+
 with gr.Blocks(title=TITLE) as demo:
     gr.Markdown(f"""# {TITLE}
 
 Space opérationnel en mode **{MODE}**.
 
-- Compatible ZeroGPU et CPU (sans dépendance CUDA obligatoire)
-- Génération de test locale pour valider le pipeline
+- Compatible ZeroGPU et CPU
+- Génération de test locale
 """)
     inp = gr.Textbox(label="Prompt", placeholder="Décris ce que tu veux générer...")
     run = gr.Button("Générer")
@@ -64,6 +74,5 @@ Space opérationnel en mode **{MODE}**.
     status = gr.Textbox(label="Statut")
     run.click(infer, inputs=inp, outputs=[media, status])
 
-
 if __name__ == "__main__":
     demo.launch()

packages.txt

@@ -0,0 +1 @@
+ffmpeg

requirements.txt

@@ -3,3 +3,4 @@ huggingface_hub<1.0
 pillow>=10
 numpy>=1.26
 imageio[ffmpeg]>=2.34
+spaces>=0.34.0