Update app.py

app.py

@@ -13,11 +13,10 @@ 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, state):
+def text_to_pano(prompt, upscale):
     input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
     output = txt2panoimg(input_data)
-    new_state = state + 1  # Update state to force refresh
-    return output, output, new_state
+    return None, None, output, output  # Clear the Pannellum component first
 
 title = """<h1 align="center">SD-T2I-360PanoImage</h1>
 <p align="center">360° Panorama Image Generation</p>
@@ -39,14 +38,11 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
     
     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, state],
-        outputs=[t2p_output, t2p_image_output, state]
+        inputs=[t2p_input, t2p_upscale],
+        outputs=[t2p_output, t2p_image_output, t2p_output, t2p_image_output]  # Clear and then update
     )
 
 demo.launch()