bla commited on
Commit
6962f1b
·
verified ·
1 Parent(s): 1ca95e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -215,12 +215,11 @@ class YOLOWorldDetector:
215
  else:
216
  classes = None
217
 
 
218
  # YOLOWorld supports text prompts
219
  results = self.model.predict(
220
  source=image,
221
- classes=classes, # Pass as a list of strings or None
222
  conf=confidence_threshold,
223
- verbose=False
224
  )
225
  except Exception as e:
226
  print(f"Error during YOLOWorld inference: {e}")
 
215
  else:
216
  classes = None
217
 
218
+ self.model.set_classes(classes)
219
  # YOLOWorld supports text prompts
220
  results = self.model.predict(
221
  source=image,
 
222
  conf=confidence_threshold,
 
223
  )
224
  except Exception as e:
225
  print(f"Error during YOLOWorld inference: {e}")