Commit
·
ea7d27e
1
Parent(s):
8591cea
Update mp4 name
Browse files
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
|
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())
|