callum-canavan commited on
Commit
ea7d27e
·
1 Parent(s): 8591cea

Update mp4 name

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -2,9 +2,9 @@ import argparse
2
  from pathlib import Path
3
 
4
  import gradio as gr
 
5
  import torch
6
  from diffusers import DiffusionPipeline
7
- from icecream import ic
8
 
9
  from visual_anagrams.views import get_views, VIEW_MAP_NAMES
10
  from visual_anagrams.samplers import sample_stage_1, sample_stage_2
@@ -62,14 +62,16 @@ def generate_content(
62
  generator=generator)
63
  save_illusion(image, views, Path(""))
64
 
 
65
  size = image.shape[-1]
66
  animate_two_view(
67
  f"sample_{size}.png",
68
  views[1],
69
  prompts[0],
70
  prompts[1],
 
71
  )
72
- return 'tmp.mp4', f"sample_{size}.png", f"sample_{size}.views.png"
73
 
74
 
75
  choices = list(VIEW_MAP_NAMES.keys())
 
2
  from pathlib import Path
3
 
4
  import gradio as gr
5
+ from icecream import ic
6
  import torch
7
  from diffusers import DiffusionPipeline
 
8
 
9
  from visual_anagrams.views import get_views, VIEW_MAP_NAMES
10
  from visual_anagrams.samplers import sample_stage_1, sample_stage_2
 
62
  generator=generator)
63
  save_illusion(image, views, Path(""))
64
 
65
+ output_name = f"illusion.mp4"
66
  size = image.shape[-1]
67
  animate_two_view(
68
  f"sample_{size}.png",
69
  views[1],
70
  prompts[0],
71
  prompts[1],
72
+ save_video_path=output_name,
73
  )
74
+ return output_name, f"sample_{size}.png", f"sample_{size}.views.png"
75
 
76
 
77
  choices = list(VIEW_MAP_NAMES.keys())