Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -77,13 +77,13 @@ def run_generate_base_video(
|
|
| 77 |
print(f"{error_message}\nDetalhes: {traceback.format_exc()}")
|
| 78 |
raise gr.Error(error_message)
|
| 79 |
|
| 80 |
-
def run_ltx_refinement(state, prompt, neg_prompt,
|
| 81 |
if not state or not state.get("low_res_latents"):
|
| 82 |
raise gr.Error("Erro: Gere um vídeo base primeiro na Etapa 1.")
|
| 83 |
try:
|
| 84 |
video_path, tensor_path = video_generation_service.generate_upscale_denoise(
|
| 85 |
latents_path=state["low_res_latents"], prompt=prompt,
|
| 86 |
-
negative_prompt=neg_prompt,
|
| 87 |
)
|
| 88 |
state["refined_video_ltx"] = video_path; state["refined_latents_ltx"] = tensor_path
|
| 89 |
return video_path, state
|
|
@@ -205,7 +205,7 @@ with gr.Blocks() as demo:
|
|
| 205 |
|
| 206 |
ltx_refine_btn.click(
|
| 207 |
fn=run_ltx_refinement,
|
| 208 |
-
inputs=[app_state, prompt_input, neg_prompt_input
|
| 209 |
outputs=[ltx_refined_video_output, app_state]
|
| 210 |
)
|
| 211 |
|
|
|
|
| 77 |
print(f"{error_message}\nDetalhes: {traceback.format_exc()}")
|
| 78 |
raise gr.Error(error_message)
|
| 79 |
|
| 80 |
+
def run_ltx_refinement(state, prompt, neg_prompt, progress=gr.Progress(track_tqdm=True)):
|
| 81 |
if not state or not state.get("low_res_latents"):
|
| 82 |
raise gr.Error("Erro: Gere um vídeo base primeiro na Etapa 1.")
|
| 83 |
try:
|
| 84 |
video_path, tensor_path = video_generation_service.generate_upscale_denoise(
|
| 85 |
latents_path=state["low_res_latents"], prompt=prompt,
|
| 86 |
+
negative_prompt=neg_prompt, seed=state["used_seed"]
|
| 87 |
)
|
| 88 |
state["refined_video_ltx"] = video_path; state["refined_latents_ltx"] = tensor_path
|
| 89 |
return video_path, state
|
|
|
|
| 205 |
|
| 206 |
ltx_refine_btn.click(
|
| 207 |
fn=run_ltx_refinement,
|
| 208 |
+
inputs=[app_state, prompt_input, neg_prompt_input],
|
| 209 |
outputs=[ltx_refined_video_output, app_state]
|
| 210 |
)
|
| 211 |
|