Update modelling_magiv2.py

modelling_magiv2.py

@@ -103,6 +103,8 @@ class Magiv2Model(PreTrainedModel):
         
     
     def assign_names_to_characters(self, images, character_bboxes, character_bank, character_clusters, eta=0.75):
+        if len(character_bank["images"]) == 0:
+            return ["Other" for bboxes_for_image in character_bboxes for bbox in bboxes_for_image]
         chapter_wide_char_embeddings = self.predict_crop_embeddings(images, character_bboxes)
         chapter_wide_char_embeddings = torch.cat(chapter_wide_char_embeddings, dim=0)
         chapter_wide_char_embeddings = torch.nn.functional.normalize(chapter_wide_char_embeddings, p=2, dim=1).cpu().numpy()