DualityAI-RebekahBogdanoff commited on
Commit
4f5c6f8
·
verified ·
1 Parent(s): 446d2a6

Update app.py

Browse files

add test iamges

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  from ultralytics import YOLO
7
 
8
  path = ['./data/0068.jpg']
9
- path = ['./data/0210.jpg']
10
 
11
  model_path = './best.pt'
12
  model = YOLO(model_path)
@@ -23,7 +23,7 @@ iface = gr.Interface(
23
  inputs=gr.components.Image(type="filepath", label="Input Image"),
24
  outputs=gr.Image(),
25
  title="Cheerios detector",
26
- examples=path,
27
  )
28
 
29
  # Launch the interface
 
6
  from ultralytics import YOLO
7
 
8
  path = ['./data/0068.jpg']
9
+ path0 = ['./data/0210.jpg']
10
 
11
  model_path = './best.pt'
12
  model = YOLO(model_path)
 
23
  inputs=gr.components.Image(type="filepath", label="Input Image"),
24
  outputs=gr.Image(),
25
  title="Cheerios detector",
26
+ examples=path, path0
27
  )
28
 
29
  # Launch the interface