Update app.py

app.py

@@ -16,7 +16,7 @@ txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float1
 def text_to_pano(prompt, upscale):
     input_data = {'prompt': prompt, 'upscale': upscale}
     output = txt2panoimg(input_data)
-    return output
+    return output, output
 
 
 
@@ -38,10 +38,13 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
         with gr.Column():
             t2p_output = Pannellum(label="Generated 360° Panorama")
     
+    with gr.Row():
+        t2p_image_output = gr.Image(label="Generated Image")
+
     t2p_generate.click(
         text_to_pano,
         inputs=[t2p_input, t2p_upscale],
-        outputs=t2p_output
+        outputs=[t2p_output, t2p_image_output]
     )
 
 demo.launch()