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 |
|