Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,6 +16,7 @@ from glob import glob
|
|
| 16 |
from pathlib import Path
|
| 17 |
from typing import Optional
|
| 18 |
|
|
|
|
| 19 |
from diffsynth import save_video, ModelManager, SVDVideoPipeline
|
| 20 |
from diffsynth import SDVideoPipeline, ControlNetConfigUnit, VideoData, save_frames
|
| 21 |
from diffsynth.extensions.RIFE import RIFESmoother
|
|
@@ -169,10 +170,12 @@ def generate(
|
|
| 169 |
|
| 170 |
|
| 171 |
examples = [
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
|
|
|
|
|
|
| 176 |
]
|
| 177 |
|
| 178 |
|
|
@@ -253,7 +256,15 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
|
|
| 253 |
submit_btn = gr.Button(value="Generate")
|
| 254 |
#stop_btn = gr.Button(value="Stop", variant="stop")
|
| 255 |
clear_btn = gr.ClearButton([media, seed, video])
|
| 256 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 257 |
selected.change(change_media, inputs=[image_in, video_in, selected], outputs=[image_in, video_in, media, prompt])
|
| 258 |
submit_event = submit_btn.click(fn=generate, inputs=[media, selected, seed, num_inference_steps, animatediff_batch_size, animatediff_stride, motion_bucket_id, fps_id, num_frames, prompt], outputs=[video, seed], api_name="video")
|
| 259 |
#stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])
|
|
|
|
| 16 |
from pathlib import Path
|
| 17 |
from typing import Optional
|
| 18 |
|
| 19 |
+
#Core functions from https://github.com/modelscope/DiffSynth-Studio
|
| 20 |
from diffsynth import save_video, ModelManager, SVDVideoPipeline
|
| 21 |
from diffsynth import SDVideoPipeline, ControlNetConfigUnit, VideoData, save_frames
|
| 22 |
from diffsynth.extensions.RIFE import RIFESmoother
|
|
|
|
| 170 |
|
| 171 |
|
| 172 |
examples = [
|
| 173 |
+
['./walking.mp4', "A woman walking on the street", "Diffutoon"],
|
| 174 |
+
['./smilegirl.mp4', "A girl stand on the grass", "Diffutoon"],
|
| 175 |
+
['./working.mp4', "A woman is doing the dishes", "Diffutoon"],
|
| 176 |
+
["./train.jpg", "", "ExVideo"],
|
| 177 |
+
["./girl.webp", "", "ExVideo"],
|
| 178 |
+
["./robo.jpg", "", "ExVideo"],
|
| 179 |
]
|
| 180 |
|
| 181 |
|
|
|
|
| 256 |
submit_btn = gr.Button(value="Generate")
|
| 257 |
#stop_btn = gr.Button(value="Stop", variant="stop")
|
| 258 |
clear_btn = gr.ClearButton([media, seed, video])
|
| 259 |
+
|
| 260 |
+
gr.Examples(
|
| 261 |
+
examples=examples,
|
| 262 |
+
inputs=[media, prompt, selected],
|
| 263 |
+
outputs=[video, seed],
|
| 264 |
+
fn=generate
|
| 265 |
+
cache_examples="lazy",
|
| 266 |
+
examples_per_page=4,
|
| 267 |
+
)
|
| 268 |
selected.change(change_media, inputs=[image_in, video_in, selected], outputs=[image_in, video_in, media, prompt])
|
| 269 |
submit_event = submit_btn.click(fn=generate, inputs=[media, selected, seed, num_inference_steps, animatediff_batch_size, animatediff_stride, motion_bucket_id, fps_id, num_frames, prompt], outputs=[video, seed], api_name="video")
|
| 270 |
#stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])
|