qubvel-hf HF Staff commited on
Commit
8ddd00a
·
1 Parent(s): 9d19ec6

Update theme, header

Browse files
Files changed (1) hide show
  1. app.py +11 -17
app.py CHANGED
@@ -300,23 +300,23 @@ def create_video_inputs() -> List[gr.components.Component]:
300
  ]
301
 
302
 
303
- def create_button_row(is_image: bool) -> List[gr.Button]:
304
- prefix = "Image" if is_image else "Video"
305
  return [
306
  gr.Button(
307
- f"{prefix} Detect Objects", variant="primary", elem_classes="action-button"
308
  ),
309
- gr.Button(f"{prefix} Clear", variant="secondary", elem_classes="action-button"),
310
  ]
311
 
312
 
313
  # Gradio interface
314
- with gr.Blocks(theme=gr.themes.Soft()) as demo:
315
  gr.Markdown(
316
  """
317
- # Real-Time Object Detection Demo
318
- Experience state-of-the-art object detection with USTC's Dfine models. Upload an image or video,
319
- provide a URL, or try an example below. Select a model and adjust the confidence threshold to see detections in real time!
 
320
  """,
321
  elem_classes="header-text",
322
  )
@@ -333,9 +333,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
333
  image_model_checkpoint,
334
  image_confidence_threshold,
335
  ) = create_image_inputs()
336
- image_detect_button, image_clear_button = create_button_row(
337
- is_image=True
338
- )
339
  with gr.Column(scale=2):
340
  image_output = gr.AnnotatedImage(
341
  label="Detection Results",
@@ -374,12 +372,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
374
  with gr.Row():
375
  with gr.Column(scale=1, min_width=300):
376
  with gr.Group():
377
- video_input, video_checkpoint, video_confidence_threshold = (
378
- create_video_inputs()
379
- )
380
- video_detect_button, video_clear_button = create_button_row(
381
- is_image=False
382
- )
383
  with gr.Column(scale=2):
384
  video_output = gr.Video(
385
  label="Detection Results",
 
300
  ]
301
 
302
 
303
+ def create_button_row() -> List[gr.Button]:
 
304
  return [
305
  gr.Button(
306
+ f"Detect Objects", variant="primary", elem_classes="action-button"
307
  ),
308
+ gr.Button(f"Clear", variant="secondary", elem_classes="action-button"),
309
  ]
310
 
311
 
312
  # Gradio interface
313
+ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
314
  gr.Markdown(
315
  """
316
+ # Object Detection Demo
317
+ Experience state-of-the-art object detection with USTC's D-Fine models.
318
+ - **Image** and **Video** modes are supported.
319
+ - Select a model and adjust the confidence threshold to see detections!
320
  """,
321
  elem_classes="header-text",
322
  )
 
333
  image_model_checkpoint,
334
  image_confidence_threshold,
335
  ) = create_image_inputs()
336
+ image_detect_button, image_clear_button = create_button_row()
 
 
337
  with gr.Column(scale=2):
338
  image_output = gr.AnnotatedImage(
339
  label="Detection Results",
 
372
  with gr.Row():
373
  with gr.Column(scale=1, min_width=300):
374
  with gr.Group():
375
+ video_input, video_checkpoint, video_confidence_threshold = create_video_inputs()
376
+ video_detect_button, video_clear_button = create_button_row()
 
 
 
 
377
  with gr.Column(scale=2):
378
  video_output = gr.Video(
379
  label="Detection Results",