checkpoint 360 app

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from PIL import Image
 from diffusers import StableDiffusionPipeline
@@ -8,7 +9,7 @@ from gradio_pannellum import Pannellum
 import io
 from huggingface_hub import snapshot_download
 from txt2panoimg import Text2360PanoramaImagePipeline
-import spaces
+
 
 # Download the model
 model_path = snapshot_download("archerfmy0831/sd-t2i-360panoimage")
@@ -28,15 +29,17 @@ def text_to_pano(prompt, upscale):
 
 
 title = """
-<div style="text-align: center;">
+<div style="text-align: left;">
     <img src="https://ant.dpu.ac.th/wp-content/uploads/2024/04/dpulogo.png" width="200"/>
-    <h1>SD-T2I-360PanoImage</h1>
-    <p>360° Panorama Image Generation</p>
+    <h4></h4>
+    <p>This Space is a playground designed to experiment with generating 360-degree images for use in Virtual Reality (VR) experiences. It utilizes A-Frame to render the images as WebXR using a Text-to-VR approach. This demonstration was presented at the MHESI Fair 2024 in Thailand. <br/> It is conducted by the College of Creative Design and Entertainment Technology, Dhurakij Pundit University, in the lab of Asst. Prof. Banyapon Poolsawas under the MIT License.</p>
+    <p>SD-T2I-360PanoImage Generate 360° Panorama Image Generation</p>
+    <p>
+        <a href="https://github.com/ArcherFMY/SD-T2I-360PanoImage/" target="_blank">[Github]</a>
+        <a href="https://huggingface.co/archerfmy0831/sd-t2i-360panoimage" target="_blank">[Models]</a>
+    </p>
 </div>
-<p><center>
-<a href="https://github.com/ArcherFMY/SD-T2I-360PanoImage/" target="_blank">[Github]</a>
-<a href="https://huggingface.co/archerfmy0831/sd-t2i-360panoimage" target="_blank">[Models]</a>
-</center></p>
+
 """
 
 def create_aframe_html(image):
@@ -108,11 +111,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     t2p_image_output.change(create_aframe_preview, inputs=[t2p_image_output], outputs=[aframe_preview])
     
-    copy_html_button.click(
-        None,
-        inputs=[output_html],
-        outputs=[],
-        _js="(html) => {navigator.clipboard.writeText(html).then(() => alert('Copied to clipboard!')).catch(err => console.error('Failed to copy: ', err));}"
-    )
+    copy_html_button.click(lambda html: gr.File.update(value=html, visible=True), inputs=[output_html], outputs=[gr.File(label="A-Frame HTML Code", visible=False)])
 
 demo.launch()