ragavsachdeva commited on
Commit
c9d0a34
·
verified ·
1 Parent(s): ff2a8fa

Update modeling_florence2.py

Browse files
Files changed (1) hide show
  1. 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]["is_essential_texts"] = essential_text_logits
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