Koni commited on
Commit
d722369
1 Parent(s): e8a3154

Changing the examples for better fit

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,7 +7,7 @@ from huggingface_hub import hf_hub_download
7
 
8
 
9
  @spaces.GPU
10
- def yolov9_inference(img_path, model_id='YOLOv9-S_X_LOCO-converted.pt', img_size=640, conf_thres=0.1, iou_thres=0.4):
11
  """
12
  Load a YOLOv9 model, configure it, perform inference on an image, and optionally adjust
13
  the input size and apply test time augmentation.
@@ -98,10 +98,10 @@ def app():
98
  examples=[
99
  [
100
  "data/forklift.jpg",
101
- "YOLOv9-E_X_LOCO-converted.pt",
102
  640,
103
- 0.4,
104
- 0.5,
105
  ],
106
  [
107
  "data/hall.jpg",
 
7
 
8
 
9
  @spaces.GPU
10
+ def yolov9_inference(img_path, model_id='YOLOv9-S_X_LOCO-converted.pt', img_size=640, conf_thres=0.3, iou_thres=0.7):
11
  """
12
  Load a YOLOv9 model, configure it, perform inference on an image, and optionally adjust
13
  the input size and apply test time augmentation.
 
98
  examples=[
99
  [
100
  "data/forklift.jpg",
101
+ "YOLOv9-S_X_LOCO-converted.pt",
102
  640,
103
+ 0.3,
104
+ 0.7,
105
  ],
106
  [
107
  "data/hall.jpg",