syedfaisalabrar commited on
Commit
a8c009b
·
verified ·
1 Parent(s): fd0f8da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ model = YOLO(model_path)
11
 
12
  def predict(image):
13
  image = np.array(image)
14
- results = model(image)
15
 
16
  detected_classes = set() # Track unique detected classes
17
  labels = []
 
11
 
12
  def predict(image):
13
  image = np.array(image)
14
+ results = model(image, conf=0.85)
15
 
16
  detected_classes = set() # Track unique detected classes
17
  labels = []