Update app.py

app.py

@@ -1,4 +1,5 @@
-import os, json, re, sys, subprocess, gc, tqdm, math, time, random, threading, spaces, numpy, torch
+import os, json, re, sys, subprocess, gc, tqdm, math, time, random, threading, spaces, torch
+import numpy as np
 import gradio as gr
 from PIL import Image, ImageOps
 from moviepy import VideoFileClip
@@ -115,7 +116,7 @@ def convert_to_gif(video_path):
     return gif_path
 
 @spaces.GPU(duration=180)
-def plex(prompt,image_input,seed_value,scale_status,rife_status,progress=gr.Progress(track_tqdm=True)):
+def plex(prompt,image_input,stips,gscale,seed_value,scale_status,rife_status,progress=gr.Progress(track_tqdm=True)):
     seed = seed_value
     if seed == -1:
         seed = random.randint(0, 2**8 - 1)
@@ -134,7 +135,7 @@ def plex(prompt,image_input,seed_value,scale_status,rife_status,progress=gr.Prog
     image  = ImageOps.exif_transpose(Image.fromarray(tensor))
     prompt = prompt.strip('"')
     generator = torch.Generator(device).manual_seed(seed) if seed else None
-    video_pt = pipe(prompt=prompt,image=image,num_videos_per_prompt=1,num_inference_steps=10,num_frames=49,use_dynamic_cfg=False,guidance_scale=7.0,generator=generator,id_vit_hidden=id_vit_hidden,id_cond=id_cond,kps_cond=kps_cond,output_type="pt",)
+    video_pt = pipe(prompt=prompt,image=image,num_videos_per_prompt=1,num_inference_steps=stips,num_frames=49,use_dynamic_cfg=False,guidance_scale=gscale,generator=generator,id_vit_hidden=id_vit_hidden,id_cond=id_cond,kps_cond=kps_cond,output_type="pt",)
     latents = video_pt.frames
     ##free_memory()
     if scale_status:
@@ -169,7 +170,7 @@ examples_images = [
 with gr.Blocks() as demo:
     gr.Markdown("""
            <div style="text-align: center; font-size: 32px; font-weight: bold; margin-bottom: 20px;">
-               ConsisID Space🤗
+               🤗ConsisID Space🤗
            </div>
            <div style="text-align: center;">
                <a href="https://huggingface.co/BestWishYsh/ConsisID">🤗 Model Hub</a> |
@@ -192,20 +193,17 @@ with gr.Blocks() as demo:
         with gr.Column():
             with gr.Accordion("IPT2V: Face Input", open=True):
                 image_input = gr.Image(label="Input Image (should contain clear face)")
-                prompt = gr.Textbox(label="Prompt (Less than 200 Words)", placeholder="Enter your prompt here", lines=5)
+                prompt = gr.Textbox(label="Prompt (Less than 200 Words)", placeholder="Enter your prompt here", lines=3)
 
             with gr.Group():
                 with gr.Column():
                     with gr.Row():
-                        seed_param = gr.Number(
-                            label="Inference Seed (Enter a positive number, -1 for random)", value=42
-                        )
+                        stips = gr.Slider(label="Steps", minimum=6, step=1, maximum=10, value=10)
+                        gscale = gr.Slider(label="Guidance scale", minimum=1, step=0.1, maximum=20, value=7.0)
+                        seed_param = gr.Slider(label="Inference Seed (Leave -1 for random)", minimum=0, step=32, maximum=2**8 - 1, value=-1)
                     with gr.Row():
-                        enable_scale = gr.Checkbox(label="Super-Resolution (720 × 480 -> 2880 × 1920)", value=False)
-                        enable_rife = gr.Checkbox(label="Frame Interpolation (8fps -> 16fps)", value=True)
-                    gr.Markdown(
-                        "✨In this demo, we use [RIFE](https://github.com/hzwer/ECCV2022-RIFE) for frame interpolation and [Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN) for upscaling(Super-Resolution)."
-                    )
+                        enable_scale = gr.Checkbox(label="Super-Resolution (720 × 480 -> 2880 × 1920) [Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN)", value=False)
+                        enable_rife = gr.Checkbox(label="Frame Interpolation (8fps -> 16fps) [RIFE](https://github.com/hzwer/ECCV2022-RIFE)", value=True)
 
             generate_button = gr.Button("🎬 Generate Video")
 
@@ -223,7 +221,7 @@ with gr.Blocks() as demo:
                 )
     generate_button.click(
         fn=plex,
-        inputs=[prompt, image_input, seed_param, enable_scale, enable_rife],
+        inputs=[prompt, image_input, stips, gscale, seed_param, enable_scale, enable_rife],
         outputs=[video_output, download_video_button, download_gif_button, seed_text],
     )