Spaces:
Running
on
Zero
Running
on
Zero
陈硕
commited on
Commit
·
ab732e1
1
Parent(s):
d55b62c
fix pipe bug
Browse files
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 |
|