Update app.py

app.py

@@ -98,23 +98,23 @@ def app():
             cache_examples='lazy',
         )
 
-gradio_app = gr.Blocks()
-with gradio_app:
-    gr.HTML(
-        """
-    <h1 style='text-align: center'>
-    YOLOv10 for early fire detection with low resource consumption
-    </h1>
-    """)
-    gr.HTML(
-        """
-        <h3 style='text-align: center'>
-        Original paper and code:
-        <a href='https://arxiv.org/abs/2405.14458' target='_blank'>arXiv</a> | <a href='https://github.com/THU-MIG/yolov10' target='_blank'>github</a>
-        </h3>
-        """)
-    with gr.Row():
-        with gr.Column():
-            app()
 if __name__ == '__main__':
+    gradio_app = gr.Blocks()
+    with gradio_app:
+        gr.HTML(
+            """
+        <h1 style='text-align: center'>
+        YOLOv10 for early fire detection with low resource consumption
+        </h1>
+        """)
+        gr.HTML(
+            """
+            <h3 style='text-align: center'>
+            Original paper and code:
+            <a href='https://arxiv.org/abs/2405.14458' target='_blank'>arXiv</a> | <a href='https://github.com/THU-MIG/yolov10' target='_blank'>github</a>
+            </h3>
+            """)
+        with gr.Row():
+            with gr.Column():
+                app()
     gradio_app.launch()