陈硕 commited on
Commit
ab732e1
·
1 Parent(s): d55b62c

fix pipe bug

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -242,6 +242,7 @@ def convert_prompt(prompt: str, image_path: str = None, retry_times: int = 3) ->
242
 
243
  @spaces.GPU
244
  def infer(
 
245
  prompt: str,
246
  orbit_type: str,
247
  image_input: str,
@@ -396,9 +397,11 @@ with gr.Blocks() as demo:
396
  seed_value,
397
  scale_status,
398
  rife_status,
 
399
  progress=gr.Progress(track_tqdm=True)
400
  ):
401
  latents, seed = infer(
 
402
  prompt,
403
  orbit_type,
404
  image_input,
@@ -437,7 +440,7 @@ with gr.Blocks() as demo:
437
 
438
  generate_button.click(
439
  generate,
440
- inputs=[prompt, orbit_type, image_in, seed_param, enable_scale, enable_rife],
441
  outputs=[video_output, download_video_button, download_gif_button, seed_text],
442
  )
443
 
 
242
 
243
  @spaces.GPU
244
  def infer(
245
+ pipe_image,
246
  prompt: str,
247
  orbit_type: str,
248
  image_input: str,
 
397
  seed_value,
398
  scale_status,
399
  rife_status,
400
+ pipe_image,
401
  progress=gr.Progress(track_tqdm=True)
402
  ):
403
  latents, seed = infer(
404
+ pipe_image,
405
  prompt,
406
  orbit_type,
407
  image_input,
 
440
 
441
  generate_button.click(
442
  generate,
443
+ inputs=[prompt, orbit_type, image_in, seed_param, enable_scale, enable_rife, pipe_image],
444
  outputs=[video_output, download_video_button, download_gif_button, seed_text],
445
  )
446