gokaygokay commited on
Commit
a9c015a
1 Parent(s): 88d1633

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,7 +16,8 @@ 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 None, None, output, output # Clear the Pannellum component first
 
20
 
21
  title = """<h1 align="center">SD-T2I-360PanoImage</h1>
22
  <p align="center">360° Panorama Image Generation</p>
@@ -42,7 +43,7 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
42
  t2p_generate.click(
43
  text_to_pano,
44
  inputs=[t2p_input, t2p_upscale],
45
- outputs=[t2p_output, t2p_image_output, t2p_output, t2p_image_output] # Clear and then update
46
  )
47
 
48
  demo.launch()
 
16
  def text_to_pano(prompt, upscale):
17
  input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
18
  output = txt2panoimg(input_data)
19
+ timestamp = int(time.time() * 1000) # Add this line
20
+ return f"{output}?t={timestamp}", output # Modify this line
21
 
22
  title = """<h1 align="center">SD-T2I-360PanoImage</h1>
23
  <p align="center">360° Panorama Image Generation</p>
 
43
  t2p_generate.click(
44
  text_to_pano,
45
  inputs=[t2p_input, t2p_upscale],
46
+ outputs=[t2p_output, t2p_image_output]
47
  )
48
 
49
  demo.launch()