Spaces:
Runtime error
Runtime error
Badr AlKhamissi
commited on
Commit
•
aaf94ae
1
Parent(s):
470b2c3
resolving gif issue
Browse files
app.py
CHANGED
@@ -8,6 +8,7 @@ import random
|
|
8 |
import numpy.random as npr
|
9 |
import sys
|
10 |
import imageio
|
|
|
11 |
|
12 |
# sys.path.append('./code')
|
13 |
|
@@ -222,7 +223,7 @@ def run_main_app(semantic_concept, word, prompt_suffix, font_name, num_steps, se
|
|
222 |
img = img[:, :, 3:4] * img[:, :, :3] + torch.ones(img.shape[0], img.shape[1], 3, device=device) * (
|
223 |
1 - img[:, :, 3:4])
|
224 |
img = img[:, :, :3]
|
225 |
-
gif_frames += [img.detach().cpu().numpy()]
|
226 |
|
227 |
|
228 |
filename = os.path.join(cfg.experiment_dir, "video-svg", f"iter{step:04d}.svg")
|
@@ -259,7 +260,7 @@ def run_main_app(semantic_concept, word, prompt_suffix, font_name, num_steps, se
|
|
259 |
combine_word(cfg.word, cfg.optimized_letter, cfg.font, cfg.experiment_dir, device)
|
260 |
|
261 |
filename = os.path.join(cfg.experiment_dir, "final.gif")
|
262 |
-
imageio.mimsave(filename, gif_frames)
|
263 |
|
264 |
yield gr.update(value=filename_init,visible=True),gr.update(visible=False),gr.update(value=filename,visible=True)
|
265 |
|
@@ -357,7 +358,7 @@ with gr.Blocks() as demo:
|
|
357 |
with gr.Column():
|
358 |
result0 = gr.Image(type="filepath", label="Initial Word").style(height=250)
|
359 |
result1 = gr.Image(type="filepath", label="Optimization Process").style(height=300)
|
360 |
-
result2 = gr.
|
361 |
|
362 |
|
363 |
with gr.Row():
|
|
|
8 |
import numpy.random as npr
|
9 |
import sys
|
10 |
import imageio
|
11 |
+
import numpy as np
|
12 |
|
13 |
# sys.path.append('./code')
|
14 |
|
|
|
223 |
img = img[:, :, 3:4] * img[:, :, :3] + torch.ones(img.shape[0], img.shape[1], 3, device=device) * (
|
224 |
1 - img[:, :, 3:4])
|
225 |
img = img[:, :, :3]
|
226 |
+
gif_frames += [img.detach().cpu().numpy()*255]
|
227 |
|
228 |
|
229 |
filename = os.path.join(cfg.experiment_dir, "video-svg", f"iter{step:04d}.svg")
|
|
|
260 |
combine_word(cfg.word, cfg.optimized_letter, cfg.font, cfg.experiment_dir, device)
|
261 |
|
262 |
filename = os.path.join(cfg.experiment_dir, "final.gif")
|
263 |
+
imageio.mimsave(filename, np.array(gif_frames).astype(np.uint8))
|
264 |
|
265 |
yield gr.update(value=filename_init,visible=True),gr.update(visible=False),gr.update(value=filename,visible=True)
|
266 |
|
|
|
358 |
with gr.Column():
|
359 |
result0 = gr.Image(type="filepath", label="Initial Word").style(height=250)
|
360 |
result1 = gr.Image(type="filepath", label="Optimization Process").style(height=300)
|
361 |
+
result2 = gr.Image(type="filepath", label="Final Result",visible=False).style(height=300)
|
362 |
|
363 |
|
364 |
with gr.Row():
|