Spaces:
Running
on
Zero
Running
on
Zero
Update gradio_app.py
Browse files- gradio_app.py +7 -4
gradio_app.py
CHANGED
@@ -103,19 +103,22 @@ def infer(frame1_path, frame2_path):
|
|
103 |
noise_injection_steps=noise_injection_steps, # 0
|
104 |
noise_injection_ratio= noise_injection_ratio, # 0.5
|
105 |
).frames[0]
|
|
|
|
|
106 |
|
107 |
out_dir = "result"
|
108 |
|
109 |
check_outputs_folder(out_dir)
|
110 |
os.makedirs(out_dir, exist_ok=True)
|
111 |
out_path = "result/video_result.gif"
|
112 |
-
|
|
|
113 |
if out_path.endswith('.gif'):
|
114 |
frames[0].save(out_path, save_all=True, append_images=frames[1:], duration=142, loop=0)
|
115 |
else:
|
116 |
export_to_video(frames, out_path, fps=7)
|
117 |
-
|
118 |
-
return
|
119 |
|
120 |
with gr.Blocks() as demo:
|
121 |
|
@@ -127,7 +130,7 @@ with gr.Blocks() as demo:
|
|
127 |
image_input2 = gr.Image(type="filepath")
|
128 |
submit_btn = gr.Button("Submit")
|
129 |
with gr.Column():
|
130 |
-
output = gr.
|
131 |
|
132 |
submit_btn.click(
|
133 |
fn = infer,
|
|
|
103 |
noise_injection_steps=noise_injection_steps, # 0
|
104 |
noise_injection_ratio= noise_injection_ratio, # 0.5
|
105 |
).frames[0]
|
106 |
+
|
107 |
+
print(f"FRAMES: {frames}")
|
108 |
|
109 |
out_dir = "result"
|
110 |
|
111 |
check_outputs_folder(out_dir)
|
112 |
os.makedirs(out_dir, exist_ok=True)
|
113 |
out_path = "result/video_result.gif"
|
114 |
+
|
115 |
+
'''
|
116 |
if out_path.endswith('.gif'):
|
117 |
frames[0].save(out_path, save_all=True, append_images=frames[1:], duration=142, loop=0)
|
118 |
else:
|
119 |
export_to_video(frames, out_path, fps=7)
|
120 |
+
'''
|
121 |
+
return "done"
|
122 |
|
123 |
with gr.Blocks() as demo:
|
124 |
|
|
|
130 |
image_input2 = gr.Image(type="filepath")
|
131 |
submit_btn = gr.Button("Submit")
|
132 |
with gr.Column():
|
133 |
+
output = gr.Textbox()
|
134 |
|
135 |
submit_btn.click(
|
136 |
fn = infer,
|