Spaces:
Running
Running
Update app.py
Browse files
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}")
|