vilarin commited on
Commit
80eaab0
·
verified ·
1 Parent(s): 56f6228

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
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
- "./train.jpg",
173
- "./girl.webp",
174
- "./robo.jpg",
175
- './working.mp4',
 
 
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])