gokaygokay commited on
Commit
a064b8e
1 Parent(s): ca23028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
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, gr.update(visible=True, value=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, visible=False)
40
 
41
  with gr.Row():
42
- t2p_image_output = gr.Image(label="Generated Image", visible=False)
43
 
44
  t2p_generate.click(
45
  text_to_pano,
46
  inputs=[t2p_input, t2p_upscale],
47
- outputs=[t2p_output, t2p_image_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()