Spaces:
Running
on
Zero
Running
on
Zero
Update gradio_tabs/animation.py
Browse files- gradio_tabs/animation.py +13 -13
gradio_tabs/animation.py
CHANGED
@@ -157,8 +157,8 @@ def clear_media():
|
|
157 |
return None, None, *([0] * len(labels_k))
|
158 |
|
159 |
|
160 |
-
image_output = gr.Image(label="Output Image",
|
161 |
-
video_output = gr.Video(label="Output Video",
|
162 |
|
163 |
|
164 |
@torch.no_grad()
|
@@ -172,7 +172,7 @@ def animation():
|
|
172 |
with gr.Column(scale=1):
|
173 |
with gr.Row():
|
174 |
with gr.Accordion(open=True, label="Source Image"):
|
175 |
-
image_input = gr.Image(type="filepath",
|
176 |
gr.Examples(
|
177 |
examples=[
|
178 |
["./data/source/macron.png"],
|
@@ -188,7 +188,7 @@ def animation():
|
|
188 |
)
|
189 |
|
190 |
with gr.Accordion(open=True, label="Driving Video"):
|
191 |
-
video_input = gr.Video(width=512,
|
192 |
gr.Examples(
|
193 |
examples=[
|
194 |
["./data/driving/driving1.mp4"],
|
@@ -207,10 +207,10 @@ def animation():
|
|
207 |
with gr.Row():
|
208 |
with gr.Column(scale=1):
|
209 |
with gr.Row(): # Buttons now within a single Row
|
210 |
-
edit_btn = gr.Button("Edit",
|
211 |
-
clear_btn = gr.Button("Clear",
|
212 |
with gr.Row():
|
213 |
-
animate_btn = gr.Button("Animate",
|
214 |
|
215 |
|
216 |
|
@@ -227,31 +227,31 @@ def animation():
|
|
227 |
with gr.Tab("Head"):
|
228 |
with gr.Row():
|
229 |
for k in labels_k[:3]:
|
230 |
-
slider = gr.Slider(minimum=-1.0, maximum=0.5, value=0, label=k,
|
231 |
inputs_s.append(slider)
|
232 |
with gr.Row():
|
233 |
for k in labels_k[3:6]:
|
234 |
-
slider = gr.Slider(minimum=-0.5, maximum=0.5, value=0, label=k,
|
235 |
inputs_s.append(slider)
|
236 |
|
237 |
with gr.Tab("Mouth"):
|
238 |
with gr.Row():
|
239 |
for k in labels_k[6:8]:
|
240 |
-
slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k,
|
241 |
inputs_s.append(slider)
|
242 |
with gr.Row():
|
243 |
for k in labels_k[8:10]:
|
244 |
-
slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k,
|
245 |
inputs_s.append(slider)
|
246 |
|
247 |
with gr.Tab("Eyes"):
|
248 |
with gr.Row():
|
249 |
for k in labels_k[10:12]:
|
250 |
-
slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k,
|
251 |
inputs_s.append(slider)
|
252 |
with gr.Row():
|
253 |
for k in labels_k[12:14]:
|
254 |
-
slider = gr.Slider(minimum=-0.2, maximum=0.2, value=0, label=k,
|
255 |
inputs_s.append(slider)
|
256 |
|
257 |
|
|
|
157 |
return None, None, *([0] * len(labels_k))
|
158 |
|
159 |
|
160 |
+
image_output = gr.Image(label="Output Image", elem_id="output_img", type='numpy', interactive=False, width=512)
|
161 |
+
video_output = gr.Video(label="Output Video", elem_id="output_vid", width=512)
|
162 |
|
163 |
|
164 |
@torch.no_grad()
|
|
|
172 |
with gr.Column(scale=1):
|
173 |
with gr.Row():
|
174 |
with gr.Accordion(open=True, label="Source Image"):
|
175 |
+
image_input = gr.Image(type="filepath", elem_id="input_img", width=512) # , height=550)
|
176 |
gr.Examples(
|
177 |
examples=[
|
178 |
["./data/source/macron.png"],
|
|
|
188 |
)
|
189 |
|
190 |
with gr.Accordion(open=True, label="Driving Video"):
|
191 |
+
video_input = gr.Video(width=512, elem_id="input_vid",) # , height=550)
|
192 |
gr.Examples(
|
193 |
examples=[
|
194 |
["./data/driving/driving1.mp4"],
|
|
|
207 |
with gr.Row():
|
208 |
with gr.Column(scale=1):
|
209 |
with gr.Row(): # Buttons now within a single Row
|
210 |
+
edit_btn = gr.Button("Edit", elem_id="button_edit",)
|
211 |
+
clear_btn = gr.Button("Clear", elem_id="button_clear")
|
212 |
with gr.Row():
|
213 |
+
animate_btn = gr.Button("Animate", elem_id="button_animate")
|
214 |
|
215 |
|
216 |
|
|
|
227 |
with gr.Tab("Head"):
|
228 |
with gr.Row():
|
229 |
for k in labels_k[:3]:
|
230 |
+
slider = gr.Slider(minimum=-1.0, maximum=0.5, value=0, label=k, elem_id="slider_"+str(k))
|
231 |
inputs_s.append(slider)
|
232 |
with gr.Row():
|
233 |
for k in labels_k[3:6]:
|
234 |
+
slider = gr.Slider(minimum=-0.5, maximum=0.5, value=0, label=k, elem_id="slider_"+str(k))
|
235 |
inputs_s.append(slider)
|
236 |
|
237 |
with gr.Tab("Mouth"):
|
238 |
with gr.Row():
|
239 |
for k in labels_k[6:8]:
|
240 |
+
slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k, elem_id="slider_"+str(k))
|
241 |
inputs_s.append(slider)
|
242 |
with gr.Row():
|
243 |
for k in labels_k[8:10]:
|
244 |
+
slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k, elem_id="slider_"+str(k))
|
245 |
inputs_s.append(slider)
|
246 |
|
247 |
with gr.Tab("Eyes"):
|
248 |
with gr.Row():
|
249 |
for k in labels_k[10:12]:
|
250 |
+
slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k, elem_id="slider_"+str(k))
|
251 |
inputs_s.append(slider)
|
252 |
with gr.Row():
|
253 |
for k in labels_k[12:14]:
|
254 |
+
slider = gr.Slider(minimum=-0.2, maximum=0.2, value=0, label=k, elem_id="slider_"+str(k))
|
255 |
inputs_s.append(slider)
|
256 |
|
257 |
|