Update app.py

app.py

@@ -40,15 +40,15 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
         t2p_image_output = gr.Image(label="Generated Image")
 
     # Add a hidden component to store a random value
-    update_trigger = gr.Number(visible=False)
+    update_trigger = gr.State(value=0)
 
-    def generate_with_update(prompt, upscale):
+    def generate_with_update(prompt, upscale, trigger):
         output, image = text_to_pano(prompt, upscale)
-        return output, image, gr.Number.update(value=torch.rand(1).item())
+        return output, image, trigger + 1
 
     t2p_generate.click(
         generate_with_update,
-        inputs=[t2p_input, t2p_upscale],
+        inputs=[t2p_input, t2p_upscale, update_trigger],
         outputs=[t2p_output, t2p_image_output, update_trigger]
     )