BestWishYsh commited on
Commit
60490d3
·
verified ·
1 Parent(s): 10998bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -196,6 +196,7 @@ with gr.Blocks() as demo:
196
 
197
  with gr.Group():
198
  with gr.Column():
 
199
  with gr.Row():
200
  seed_param = gr.Number(
201
  label="Inference Seed (Enter a positive number, -1 for random)", value=42
@@ -288,6 +289,7 @@ with gr.Blocks() as demo:
288
  prompt,
289
  negative_prompt,
290
  image_input,
 
291
  seed_value,
292
  cfg_param,
293
  scale_status,
@@ -298,7 +300,7 @@ with gr.Blocks() as demo:
298
  prompt,
299
  image_input,
300
  negative_prompt=negative_prompt,
301
- num_inference_steps=50,
302
  guidance_scale=cfg_param,
303
  seed=seed_value,
304
  scale_status=scale_status,
@@ -315,7 +317,7 @@ with gr.Blocks() as demo:
315
 
316
  generate_button.click(
317
  fn=run,
318
- inputs=[prompt, negative_prompt, image_input, seed_param, cfg_param, enable_scale, enable_rife],
319
  outputs=[video_output, download_video_button, download_gif_button, seed_text],
320
  )
321
 
 
196
 
197
  with gr.Group():
198
  with gr.Column():
199
+ num_inference_steps = gr.Slider(1, 100, value=50, step=1, label="Number of Inference Steps")
200
  with gr.Row():
201
  seed_param = gr.Number(
202
  label="Inference Seed (Enter a positive number, -1 for random)", value=42
 
289
  prompt,
290
  negative_prompt,
291
  image_input,
292
+ num_inference_steps,
293
  seed_value,
294
  cfg_param,
295
  scale_status,
 
300
  prompt,
301
  image_input,
302
  negative_prompt=negative_prompt,
303
+ num_inference_steps=num_inference_steps,
304
  guidance_scale=cfg_param,
305
  seed=seed_value,
306
  scale_status=scale_status,
 
317
 
318
  generate_button.click(
319
  fn=run,
320
+ inputs=[prompt, negative_prompt, image_input, num_inference_steps, seed_param, cfg_param, enable_scale, enable_rife],
321
  outputs=[video_output, download_video_button, download_gif_button, seed_text],
322
  )
323