Change imports

modeling_vatrpp.py

@@ -10,6 +10,7 @@ from .data.dataset import FolderDataset
 from .models.model import VATr
 from .models.util.vision import detect_text_bounds
 from torchvision.transforms.functional import to_pil_image
+from huggingface_hub import hf_hub_download
 
 
 def get_long_tail_chars():
@@ -26,6 +27,10 @@ class VATrPP(PreTrainedModel):
 
     def __init__(self, config: VATrPPConfig) -> None:
         super().__init__(config)
+
+        config.english_words_path = hf_hub_download(repo_id="blowing-up-groundhogs/vatrpp", filename=config.english_words_path)
+        config.mytext_path = hf_hub_download(repo_id="blowing-up-groundhogs/vatrpp", filename='mytext.txt')
+
         self.model = VATr(config)
         self.model.eval()
 

models/model.py

@@ -260,7 +260,7 @@ class VATr(nn.Module):
 
         self.epoch = 0
 
-        with open('mytext.txt', 'r', encoding='utf-8') as f:
+        with open(args.mytext_path, 'r', encoding='utf-8') as f:
             self.text = f.read()
             self.text = self.text.replace('\n', ' ')
             self.text = self.text.replace('\n', ' ')