vincentb25 commited on
Commit
5390360
·
1 Parent(s): de3e659

Added smaller model and changed examples

Browse files
models/{best.pt → yolo11m.pt} RENAMED
File without changes
models/yolo11s.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:540894613f3563a4537865c39131125003c0cccde281cd4bb16fc86ce36f4b2c
3
+ size 19222483
yolo-inference-project/src/app.py CHANGED
@@ -20,10 +20,10 @@ def get_model_names():
20
 
21
  model_names = get_model_names()
22
 
23
- image_paths= [['examples/smaller_many_cans.jpg', 'best.pt', 0.5],
24
- ['examples/gazebo_all.jpg', 'best.pt', 0.5],
25
- ['examples/sequence_unity.jpg', 'best.pt', 0.5],
26
- ['examples/real_plank.jpg', 'best.pt', 0.5],
27
  ]
28
 
29
  demo = gr.Interface(
@@ -37,7 +37,7 @@ demo = gr.Interface(
37
  title="YOLO Model Inference",
38
  description="Select a YOLO model, upload an image, and set the confidence threshold to perform inference.",
39
  examples=image_paths,
40
- flagging_mode="auto"
41
  )
42
 
43
  demo.launch()
 
20
 
21
  model_names = get_model_names()
22
 
23
+ image_paths= [['examples/smaller_many_cans.jpg', 'yolo11m.pt', 0.5],
24
+ ['examples/gazebo_all.jpg', 'yolo11m.pt', 0.5],
25
+ ['examples/sequence_unity.jpg', 'yolo11m.pt', 0.5],
26
+ ['examples/real_plank.jpg', 'yolo11m.pt', 0.5],
27
  ]
28
 
29
  demo = gr.Interface(
 
37
  title="YOLO Model Inference",
38
  description="Select a YOLO model, upload an image, and set the confidence threshold to perform inference.",
39
  examples=image_paths,
40
+ # flagging_mode="auto"
41
  )
42
 
43
  demo.launch()