Update app.py

app.py

@@ -13,10 +13,11 @@ model_path = snapshot_download("archerfmy0831/sd-t2i-360panoimage")
 txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float16)
 
 @spaces.GPU(duration=200)
-def text_to_pano(prompt, upscale):
+def text_to_pano(prompt, upscale, state):
     input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
     output = txt2panoimg(input_data)
-    return output, output
+    new_state = state + 1  # Update state to force refresh
+    return output, output, new_state
 
 title = """<h1 align="center">SD-T2I-360PanoImage</h1>
 <p align="center">360° Panorama Image Generation</p>
@@ -34,15 +35,18 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
             t2p_upscale = gr.Checkbox(label="Upscale (takes about 60 seconds 6144x3072 resolution)")
             t2p_generate = gr.Button("Generate Panorama")
         with gr.Column(variant="default"):
-            t2p_output = Pannellum(show_label=False, interactive=True, key="pano_output")
+            t2p_output = Pannellum(show_label=False, interactive=True)
     
     with gr.Row():
         t2p_image_output = gr.Image(label="Generated Image")
+    
+    # Hidden state to force refresh
+    state = gr.State(value=0)
 
     t2p_generate.click(
         text_to_pano,
-        inputs=[t2p_input, t2p_upscale],
-        outputs=[t2p_output, t2p_image_output]
+        inputs=[t2p_input, t2p_upscale, state],
+        outputs=[t2p_output, t2p_image_output, state]
     )
 
 demo.launch()