Spaces:
Runtime error
Runtime error
Commit
•
da20a6b
1
Parent(s):
92158ba
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ def detect_objects(model_name, image_input, threshold, display_mask=False, img_i
|
|
| 37 |
processed_outputs,
|
| 38 |
decoder_attention_map,
|
| 39 |
encoder_attention_map,
|
| 40 |
-
cross_attention_map,
|
| 41 |
) = make_prediction(image_input, feature_extractor, model)
|
| 42 |
|
| 43 |
viz_img = visualize_prediction(
|
|
|
|
| 37 |
processed_outputs,
|
| 38 |
decoder_attention_map,
|
| 39 |
encoder_attention_map,
|
| 40 |
+
# cross_attention_map,
|
| 41 |
) = make_prediction(image_input, feature_extractor, model)
|
| 42 |
|
| 43 |
viz_img = visualize_prediction(
|