Update app.py

app.py

@@ -120,34 +120,37 @@ css = """
 #     button_primary_background_fill="*primary_500",  # 按鈕顏色（橘色）
 #     body_text_color="*neutral_200",  # 文字顏色（淺色）
 # )
-with gr.Blocks(css=css, theme='gstaff/xkcd') as demo:
-    with gr.Column():
-        gr.HTML(
-            value=(
-                '<div class="main-title">Litton7景觀分類模型</div>'
-                '<div class="reference">引用資料：'
-                '<a href="https://www.airitilibrary.com/Article/Detail/10125434-N202406210003-00003" target="_blank">'
-                "何立智、李沁築、邱浩修(2024)。Litton7：Litton視覺景觀分類深度學習模型。戶外遊憩研究，37(2)"
-                "</a>"
-                "</div>"
-            ),
-        )
-
-        with gr.Row(equal_height=True):
-            image_input = gr.Image(label="上傳影像", type="pil")
-            bar_chart = gr.Plot(label="分類結果")
-        start_button = gr.Button("開始分類", variant="primary")
-
-        start_button.click(
-            fn=Classifier().predict,
-            inputs=image_input,
-            outputs=bar_chart,
-        )
-
-        gr.HTML(
-            '<div class="footer">© 2024 LCL 版權所有<br>開發者：何立智、楊哲睿</div>',
-        )
+def get_layout():
+    with gr.Blocks(css=None, theme='gstaff/xkcd') as demo:
+        with gr.Column():
+            gr.HTML(
+                value=(
+                    '<div class="main-title">Litton7景觀分類模型</div>'
+                    '<div class="reference">引用資料：'
+                    '<a href="https://www.airitilibrary.com/Article/Detail/10125434-N202406210003-00003" target="_blank">'
+                    "何立智、李沁築、邱浩修(2024)。Litton7：Litton視覺景觀分類深度學習模型。戶外遊憩研究，37(2)"
+                    "</a>"
+                    "</div>"
+                ),
+            )
+    
+            with gr.Row(equal_height=True):
+                image_input = gr.Image(label="上傳影像", type="pil")
+                bar_chart = gr.Plot(label="分類結果")
+            start_button = gr.Button("開始分類", variant="primary")
+    
+            start_button.click(
+                fn=Classifier().predict,
+                inputs=image_input,
+                outputs=bar_chart,
+            )
+    
+            gr.HTML(
+                '<div class="footer">© 2024 LCL 版權所有<br>開發者：何立智、楊哲睿</div>',
+            )
+
+        return demo
 
 
 if __name__ == "__main__":
-    demo.queue().launch(server_name="0.0.0.0")
+    get_layout().queue().launch(server_name="0.0.0.0")