Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -16,9 +16,7 @@ txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float1
|
|
16 |
def text_to_pano(prompt, upscale):
|
17 |
input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
|
18 |
output = txt2panoimg(input_data)
|
19 |
-
return output, output
|
20 |
-
|
21 |
-
|
22 |
|
23 |
title = """<h1 align="center">SD-T2I-360PanoImage</h1>
|
24 |
<p align="center">360° Panorama Image Generation</p>
|
@@ -36,15 +34,15 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
|
|
36 |
t2p_upscale = gr.Checkbox(label="Upscale (takes about 60 seconds 6144x3072 resolution)")
|
37 |
t2p_generate = gr.Button("Generate Panorama")
|
38 |
with gr.Column(variant="default"):
|
39 |
-
t2p_output = Pannellum(show_label=False, interactive=True,
|
40 |
|
41 |
with gr.Row():
|
42 |
-
t2p_image_output = gr.Image(label="Generated Image"
|
43 |
|
44 |
t2p_generate.click(
|
45 |
text_to_pano,
|
46 |
inputs=[t2p_input, t2p_upscale],
|
47 |
-
outputs=[t2p_output, t2p_image_output
|
48 |
)
|
49 |
|
50 |
demo.launch()
|
|
|
16 |
def text_to_pano(prompt, upscale):
|
17 |
input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
|
18 |
output = txt2panoimg(input_data)
|
19 |
+
return output, output
|
|
|
|
|
20 |
|
21 |
title = """<h1 align="center">SD-T2I-360PanoImage</h1>
|
22 |
<p align="center">360° Panorama Image Generation</p>
|
|
|
34 |
t2p_upscale = gr.Checkbox(label="Upscale (takes about 60 seconds 6144x3072 resolution)")
|
35 |
t2p_generate = gr.Button("Generate Panorama")
|
36 |
with gr.Column(variant="default"):
|
37 |
+
t2p_output = Pannellum(show_label=False, interactive=True, key="pano_output")
|
38 |
|
39 |
with gr.Row():
|
40 |
+
t2p_image_output = gr.Image(label="Generated Image")
|
41 |
|
42 |
t2p_generate.click(
|
43 |
text_to_pano,
|
44 |
inputs=[t2p_input, t2p_upscale],
|
45 |
+
outputs=[t2p_output, t2p_image_output]
|
46 |
)
|
47 |
|
48 |
demo.launch()
|