Update app.py

app.py

@@ -12,6 +12,7 @@ from ultralytics.yolo.data import utils
 
 model = YOLO('yolov8n.pt')
 
+
 #load class_names
 yaml_path = str(Path(ultralytics.__file__).parent/'datasets/coco128.yaml') 
 class_names = utils.yaml_load(yaml_path)['names']
@@ -27,33 +28,35 @@ def detect(img):
         vis = img
     return vis
 
-with gr.Interface(fn=lambda x: x) as iface:
-    gr.Markdown("# Willkommen zu meiner App!")  # Dieser Text wird oberhalb der Tabs angezeigt.
-
-    with gr.Blocks() as demo:
-        with gr.Tab("Upload"):
-            input_img = gr.Image(type='pil')
-            button = gr.Button("Detect",variant="primary")
-            
-            gr.Markdown("## Output")
-            out_img = gr.Image(type='pil')
-            
-            button.click(detect,
-                         inputs=input_img, 
-                         outputs=out_img)
+with gr.Blocks() as demo:
+
+    header_html = gr.Html("<h1>foodServed, drinkServed, person, V0.0.1</h1>")
+
+    with gr.Tab("Upload"):
+        header_html.display()
+        input_img = gr.Image(type='pil')
+        button = gr.Button("Detect",variant="primary")
+        
+        gr.Markdown("## Output")
+        out_img = gr.Image(type='pil')
         
-        with gr.Tab("Url"):
-            default_url = 'https://t7.baidu.com/it/u=3601447414,1764260638&fm=193&f=GIF'
-            url = gr.Textbox(value=default_url)
-            button = gr.Button("Detect",variant="primary")
-            
-            gr.Markdown("## Output")
-            out_img = gr.Image(type='pil')
-            
-            button.click(detect,
-                         inputs=url, 
-                         outputs=out_img)
-
-    demo.launch()
-
-gr.close_all() 
+        button.click(detect,
+                     inputs=input_img, 
+                     outputs=out_img)
+
+    with gr.Tab("Url"):
+        header_html.display()
+        default_url = 'https://t7.baidu.com/it/u=3601447414,1764260638&fm=193&f=GIF'
+        url = gr.Textbox(value=default_url)
+        button = gr.Button("Detect",variant="primary")
+        
+        gr.Markdown("## Output")
+        out_img = gr.Image(type='pil')
+        
+        button.click(detect,
+                     inputs=url, 
+                     outputs=out_img)
+
+gr.close_all() 
+demo.queue()
+demo.launch()