linoyts HF Staff commited on
Commit
15e6d77
·
verified ·
1 Parent(s): 63481f2

add examples

Browse files
Files changed (1) hide show
  1. app.py +31 -1
app.py CHANGED
@@ -86,7 +86,16 @@ stream = AsyncStream()
86
  outputs_folder = './outputs/'
87
  os.makedirs(outputs_folder, exist_ok=True)
88
 
 
 
 
 
 
 
 
 
89
 
 
90
  @torch.no_grad()
91
  def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf):
92
  total_latent_sections = (total_second_length * 30) / (latent_window_size * 4)
@@ -288,7 +297,20 @@ def get_duration(input_image, prompt, t2v, n_prompt, seed, total_second_length,
288
  return total_second_length * 60
289
 
290
  @spaces.GPU(duration=get_duration)
291
- def process(input_image, prompt, t2v, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf):
 
 
 
 
 
 
 
 
 
 
 
 
 
292
  global stream
293
 
294
  # assert input_image is not None, 'No input image!'
@@ -385,5 +407,13 @@ adapted from the officical code repo [FramePack](https://github.com/lllyasviel/F
385
  start_button.click(fn=process, inputs=ips, outputs=[result_video, preview_image, progress_desc, progress_bar, start_button, end_button])
386
  end_button.click(fn=end_process)
387
 
 
 
 
 
 
 
 
 
388
 
389
  block.launch(share=True)
 
86
  outputs_folder = './outputs/'
87
  os.makedirs(outputs_folder, exist_ok=True)
88
 
89
+ examples = [
90
+ ["img_examples/1.png", "The girl dances gracefully, with clear movements, full of charm.",],
91
+ ["img_examples/2.jpg", "The man dances flamboyantly, swinging his hips and striking bold poses with dramatic flair."],
92
+ ["img_examples/3.png", "The woman dances elegantly among the blossoms, spinning slowly with flowing sleeves and graceful hand movements."],
93
+ ]
94
+
95
+ def generate_examples(input_image, prompt):
96
+ return process(input_image, prompt)
97
 
98
+
99
  @torch.no_grad()
100
  def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf):
101
  total_latent_sections = (total_second_length * 30) / (latent_window_size * 4)
 
297
  return total_second_length * 60
298
 
299
  @spaces.GPU(duration=get_duration)
300
+ def process(input_image, prompt,
301
+ t2v=False,
302
+ n_prompt="",
303
+ seed=31337,
304
+ total_second_length=5,
305
+ latent_window_size=9,
306
+ steps=25,
307
+ cfg=1.0,
308
+ gs=10.0,
309
+ rs=0.0,
310
+ gpu_memory_preservation=6,
311
+ use_teacache=True,
312
+ mp4_crf=16
313
+ ):
314
  global stream
315
 
316
  # assert input_image is not None, 'No input image!'
 
407
  start_button.click(fn=process, inputs=ips, outputs=[result_video, preview_image, progress_desc, progress_bar, start_button, end_button])
408
  end_button.click(fn=end_process)
409
 
410
+ gr.Examples(
411
+ examples,
412
+ inputs=[input_image, prompt],
413
+ outputs=[result_video, preview_image, progress_desc, progress_bar, start_button, end_button],
414
+ fn=generate_examples,
415
+ cache_examples=True
416
+ )
417
+
418
 
419
  block.launch(share=True)