Upload anytext.py

text_embedding_module/anytext.py

@@ -471,6 +471,11 @@ class TextEmbeddingModule(nn.Module):
         args["rec_image_shape"] = "3, 48, 320"
         args["rec_batch_num"] = 6
         args["rec_char_dict_path"] = "./text_embedding_module/OCR/ppocr_keys_v1.txt"
+        args["rec_char_dict_path"] = hf_hub_download(
+            repo_id="tolgacangoz/anytext",
+            filename="text_embedding_module/OCR/ppocr_keys_v1.txt",
+            cache_dir=HF_MODULES_CACHE
+        )
         args["use_fp16"] = use_fp16
         self.embedding_manager.recog = TextRecognizer(args, self.text_predictor)