Change imports

models/unifont_module.py

@@ -2,7 +2,7 @@ import math
 import os
 
 import cv2
-import matplotlib.pyplot as plt
+from huggingface_hub import hf_hub_download
 import torch
 import pickle
 import numpy as np
@@ -27,7 +27,14 @@ class UnifontModule(torch.nn.Module):
             self.linear = torch.nn.Identity()
 
     def get_symbols(self, input_type):
-        with open(f"files/{input_type}.pickle", "rb") as f:
+
+        file_path = hf_hub_download(
+            repo_id="blowing-up-groundhogs/vatrpp",
+            filename=f"files/{input_type}.pickle",
+            cache_dir="./hf_cache"  # Opzionale: specifica una cartella di cache
+        )
+
+        with open(file_path, "rb") as f:
             symbols = pickle.load(f)
 
         all_symbols = {sym['idx'][0]: sym['mat'].astype(np.float32) for sym in symbols}