Update app.py

app.py

@@ -15,11 +15,6 @@ SIGNS_CLASSES = ('A10', 'A11', 'A12', 'A14', 'A15', 'A16', 'A17', 'A18', 'A19',
 config_file = 'configs/config_cascade_rcnn_traffic_signs.py'
 checkpoint_file = 'checkpoints/traffic_signs_cascade_2v2.pth'
 
-with gr.Blocks() as demo:
-    gr.Markdown('''
-        # Test
-    ''')
-
 def draw_coco_bboxes(img, bboxes, color=(255,255,0), width=5, show=False, export_p=None, 
                      labels=None, resize_to=None):
 
@@ -58,4 +53,10 @@ def traffic_sign_inference(img):
     return img_res_vis.permute(1, 2, 0).numpy()
 
 demo = gr.Interface(traffic_sign_inference, gr.Image(), "image")
+
+with demo:
+    gr.Markdown('''
+        # Test
+    ''')
+
 demo.launch()