Update modeling_florence2.py
Browse files- modeling_florence2.py +1 -1
modeling_florence2.py
CHANGED
|
@@ -2791,7 +2791,7 @@ class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
|
|
| 2791 |
results[img_idx]["character_cluster_labels"] = character_cluster_labels
|
| 2792 |
results[img_idx]["text_character_associations"] = text_character_association
|
| 2793 |
results[img_idx]["text_tail_associations"] = text_tail_association
|
| 2794 |
-
results[img_idx]["
|
| 2795 |
|
| 2796 |
return results
|
| 2797 |
|
|
|
|
| 2791 |
results[img_idx]["character_cluster_labels"] = character_cluster_labels
|
| 2792 |
results[img_idx]["text_character_associations"] = text_character_association
|
| 2793 |
results[img_idx]["text_tail_associations"] = text_tail_association
|
| 2794 |
+
results[img_idx]["is_essential_text"] = essential_text_logits
|
| 2795 |
|
| 2796 |
return results
|
| 2797 |
|