Update app.py
Browse filesIncrease threshold
app.py
CHANGED
|
@@ -54,7 +54,7 @@ def detect_objects(image_url, uploaded_image, text_prompt):
|
|
| 54 |
results = processor.post_process_grounded_object_detection(
|
| 55 |
outputs,
|
| 56 |
inputs.input_ids,
|
| 57 |
-
threshold=0.
|
| 58 |
target_sizes=[image.size[::-1]]
|
| 59 |
)
|
| 60 |
|
|
|
|
| 54 |
results = processor.post_process_grounded_object_detection(
|
| 55 |
outputs,
|
| 56 |
inputs.input_ids,
|
| 57 |
+
threshold=0.4,
|
| 58 |
target_sizes=[image.size[::-1]]
|
| 59 |
)
|
| 60 |
|