Update app.py

app.py

@@ -16,28 +16,28 @@ class T2VReq(BaseModel):
     text: str = Field(...)
     negative: Optional[str] = None
     seed: Optional[int] = None
-    steps: Optional[int] = 20
-    cfg: Optional[float] = 5.0
-    width: Optional[int] = 1280
-    height: Optional[int] = 704
-    length: Optional[int] = 121
-    fps: Optional[int] = 24
+    steps: Optional[int] = 4
+    cfg: Optional[float] = 1
+    width: Optional[int] = 640
+    height: Optional[int] = 640
+    length: Optional[int] = 81
+    fps: Optional[int] = 16
     filename_prefix: Optional[str] = "video/ComfyUI"
 
 def _inject_params(prompt: Dict[str, Any], r: T2VReq) -> Dict[str, Any]:
     p = json.loads(json.dumps(prompt))
-    p["6"]["inputs"]["text"] = r.text
-    if r.seed is None:
-        r.seed = int.from_bytes(os.urandom(8), "big") & ((1 << 63) - 1)
-    p["3"]["inputs"]["seed"] = r.seed
-    if r.steps is not None: p["3"]["inputs"]["steps"] = r.steps
-    if r.cfg   is not None: p["3"]["inputs"]["cfg"]   = r.cfg
-    if r.width  is not None: p["55"]["inputs"]["width"]  = r.width
-    if r.height is not None: p["55"]["inputs"]["height"] = r.height
-    if r.length is not None: p["55"]["inputs"]["length"] = r.length
-    if r.fps is not None: p["57"]["inputs"]["fps"] = r.fps
+    p["89"]["inputs"]["text"] = r.text
+    # if r.seed is None:
+    #     r.seed = int.from_bytes(os.urandom(8), "big") & ((1 << 63) - 1)
+    # p["3"]["inputs"]["seed"] = r.seed
+    # if r.steps is not None: p["78"]["inputs"]["steps"] = r.steps
+    # if r.cfg   is not None: p["78"]["inputs"]["cfg"]   = r.cfg
+    if r.width  is not None: p["74"]["inputs"]["width"]  = r.width
+    if r.height is not None: p["74"]["inputs"]["height"] = r.height
+    if r.length is not None: p["74"]["inputs"]["length"] = r.length
+    if r.fps is not None: p["88"]["inputs"]["fps"] = r.fps
     if r.filename_prefix:
-        p["58"]["inputs"]["filename_prefix"] = r.filename_prefix
+        p["80"]["inputs"]["filename_prefix"] = r.filename_prefix
     return p
 
 def _open_ws(client_id: str, token: str):
@@ -78,10 +78,10 @@ def _extract_video_from_history(history: Dict[str, Any]) -> Dict[str, str]:
     raise RuntimeError("No video file found in history outputs")
 
 sample_prompts = [
-    "A golden retriever running across a beach at sunset, cinematic, 24fps",
-    "Aerial shot, warm colors, extreme wide shot, sunny lighting, hard lighting, daylight, establishing shot.In a desolate desert, a black SUV speeds along a highway. In a high-angle shot, the vehicle is seen driving on the left side of the road, with a roof rack and a red taillight on top. The camera slowly pushes in. In front of the vehicle are vast yellow sand dunes, and a few mountain peaks can be seen in the distance. The sky is a pale blue, and sunlight filters through the clouds, bringing a touch of warmth to the desolate land. The sides of the road are lined with dry grass dotted with some low shrubs.",
-    "Pixel art style. In a colorful universe, a player-controlled pixel character travels between planets of various shapes and unique color tones, each with strange terrain and alien creatures. A close-up shot shows the player character in the center of the frame, in dialogue with a friendly alien creature. The alien has a rounded body and large eyes, appearing very cute. Above, pixelated cosmic storms and energy vortex effects rotate slowly, adding a sense of dynamism. The overall style is retro yet futuristic, with a vibrant and lively color palette.",
-    "In an oil painting style, a vast sea of sunflowers unfolds, their golden heads blooming brilliantly in the faint light of dawn or dusk. The impasto technique is prominently used, with bold, thick brushstrokes lending a powerful texture and vitality to the sky and petals. The deep blue of the sky creates a striking contrast with the bright sunflowers, cultivating an atmosphere that is both serene and hopeful. Swarms of bees dance and hover ceaselessly among the flowers, busily collecting nectar. These dynamic bees inject boundless energy and a sense of motion into the otherwise static painting, filling the entire scene with the vibrant life of summer and the joy of harvest."
+    "A golden retriever running across a beach at sunset, cinematic",
+    "A cyberpunk city street at night with neon lights, light rain, slow pan",
+    "An astronaut walking on an alien planet covered in glowing crystals, purple sky with two moons, dust particles floating, slow panning shot, highly detailed, cinematic atmosphere.",
+    "A cat gracefully jumping between rooftops in slow motion, warm sunset lighting, camera tracking the cat midair, cinematic composition, natural movement."
 ]
 
 with gr.Blocks(
@@ -89,24 +89,22 @@ with gr.Blocks(
     theme=gr.themes.Soft(primary_hue="blue", secondary_hue="blue", neutral_hue="slate"),
 ) as demo:
     st_token = gr.State()
-    gr.Markdown("# Wan2.2 T2V running on AMD MI300x")
+    gr.Markdown("# Experience Wan2.2 14B Text-to-Video on AMD MI300X — Free Trial")
+    gr.Markdown("Powered by [AMD Devcloud](https://oneclickamd.ai/) and [ComfyUI](https://github.com/comfyanonymous/ComfyUI)")
     gr.Markdown("### Prompt")
     text = gr.Textbox(label="Prompt", placeholder="Describe the video you want", lines=3)
     
     gr.Examples(examples=sample_prompts, inputs=text)
-    gr.Markdown(
-        "[More Prompts](https://alidocs.dingtalk.com/i/nodes/EpGBa2Lm8aZxe5myC99MelA2WgN7R35y)"
-    )
 
     with gr.Accordion("Advanced Settings", open=False):
         with gr.Row():
-            width = gr.Number(label="Width", value=1280, precision=0)
-            height = gr.Number(label="Height", value=704, precision=0)
+            width = gr.Number(label="Width", value=640, precision=0)
+            height = gr.Number(label="Height", value=640, precision=0)
         with gr.Row():
-            length = gr.Number(label="Frames", value=121, precision=0)
-            fps = gr.Number(label="FPS", value=24, precision=0)
+            length = gr.Number(label="Frames", value=81, precision=0)
+            fps = gr.Number(label="FPS", value=16, precision=0)
         with gr.Row():
-            steps = gr.Number(label="Steps", value=20, precision=0)
+            steps = gr.Number(label="Steps", value=4, precision=0)
             cfg = gr.Number(label="CFG", value=5.0)
             seed = gr.Number(label="Seed (optional)", value=None)
         filename_prefix = gr.Textbox(label="Filename prefix", value="video/ComfyUI")