phillipinseoul
commited on
Commit
•
8f04382
1
Parent(s):
12a81fb
fix output
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ An interactive demo for text-guided panorama generation.
|
|
4 |
"""
|
5 |
import os
|
6 |
from os.path import join
|
|
|
7 |
import torch
|
8 |
import gradio as gr
|
9 |
|
@@ -78,9 +79,9 @@ if __name__=="__main__":
|
|
78 |
],
|
79 |
inputs=[prompt, width, sync_weight, sync_thres, seed],
|
80 |
outputs=[
|
81 |
-
join(os.path.dirname(__file__), "assets", "result_castle_seed_1.png"),
|
82 |
-
join(os.path.dirname(__file__), "assets", "result_natural_seed_2.png"),
|
83 |
-
join(os.path.dirname(__file__), "assets", "result_northern_seed_6.png"),
|
84 |
]
|
85 |
)
|
86 |
|
|
|
4 |
"""
|
5 |
import os
|
6 |
from os.path import join
|
7 |
+
from PIL import Image
|
8 |
import torch
|
9 |
import gradio as gr
|
10 |
|
|
|
79 |
],
|
80 |
inputs=[prompt, width, sync_weight, sync_thres, seed],
|
81 |
outputs=[
|
82 |
+
[gr.Image(Image.open(join(os.path.dirname(__file__), "assets", "result_castle_seed_1.png")))],
|
83 |
+
[gr.Image(Image.open(join(os.path.dirname(__file__), "assets", "result_natural_seed_2.png")))],
|
84 |
+
[gr.Image(Image.open(join(os.path.dirname(__file__), "assets", "result_northern_seed_6.png")))],
|
85 |
]
|
86 |
)
|
87 |
|