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 |
|