Update app.py

app.py

@@ -4,7 +4,6 @@ from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDisc
 from PIL import Image, ImageOps
 
 # 1. Model Setup
-# 加載 Instruct-Pix2Pix 模型
 model_id = "timbrooks/instruct-pix2pix"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -20,13 +19,11 @@ pipe.to(device)
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
 # 2. Core Function
-# 定義圖像編輯函數
 def edit_image(input_image, instruction, steps, guidance_scale, image_guidance_scale):
     if input_image is None:
         return None
     
     # Resize image for speed optimization on free tier
-    # 為了在免費服務器上運行流暢，將圖片限制在 512px
     width, height = input_image.size
     max_dim = 512
     if width > max_dim or height > max_dim:
@@ -46,8 +43,16 @@ def edit_image(input_image, instruction, steps, guidance_scale, image_guidance_s
     return images[0]
 
 # 3. Create Gradio Interface
-# 創建網頁界面 (英文版)
 with gr.Blocks(theme='gradio/soft') as demo:
+    
+    gr.HTML("""
+        <div style="padding-bottom: 10px;">
+            <a href="https://cyc1e5phere.netlify.app" target="_self" style="text-decoration: none; color: #2e7d32; font-weight: bold; font-size: 1.1em; display: flex; align-items: center; transition: opacity 0.3s;">
+                <span style="margin-right: 8px; font-size: 1.5em;">&#8592;</span> Back to Home
+            </a>
+        </div>
+    """)
+
     gr.Markdown("# 🎨 CycleSphere")
     gr.Markdown("Upload an image and enter an instruction to edit it (e.g., 'Turn the sky red' or 'Make it look like a painting').")