Update app.py

app.py

@@ -50,12 +50,6 @@ class Classifier:
             logits = self.model(input_tensor)
             probs = F.softmax(logits[:, :7], dim=1).cpu()
 
-        # probs = pd.DataFrame(
-        #     {
-        #         "class": self.LABELS,
-        #         "probs": probs[0] * 100,
-        #     }
-        # )
         return draw_bar_chart(
             {
                 "class": self.LABELS,
@@ -117,28 +111,28 @@ def get_layout():
         font-size: 14px;
     }
     """
-    theme = gr.themes.Base(
-        primary_hue="orange",
-        secondary_hue="orange",
-        neutral_hue="gray",
-        font=gr.themes.GoogleFont("Source Sans Pro"),
-    ).set(
-        background_fill_primary="*neutral_950",  # 主背景色（深黑）
-        button_primary_background_fill="*primary_500",  # 按鈕顏色（橘色）
-        body_text_color="*neutral_200",  # 文字顏色（淺色）
-    )
+    # theme = gr.themes.Base(
+    #     primary_hue="orange",
+    #     secondary_hue="orange",
+    #     neutral_hue="gray",
+    #     font=gr.themes.GoogleFont("Source Sans Pro"),
+    # ).set(
+    #     background_fill_primary="*neutral_950",  # 主背景色（深黑）
+    #     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>"
-                ),
-            )
+            # 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")
@@ -151,9 +145,9 @@ def get_layout():
                 outputs=bar_chart,
             )
     
-            gr.HTML(
-                '<div class="footer">© 2024 LCL 版權所有<br>開發者：何立智、楊哲睿</div>',
-            )
+            # gr.HTML(
+            #     '<div class="footer">© 2024 LCL 版權所有<br>開發者：何立智、楊哲睿</div>',
+            # )
 
     return demo