Remove problematic imports and add utility files to config.json auto_map

config.json

@@ -13,6 +13,11 @@
   "auto_map": {
     "AutoConfig": "configuration_ablang2paired.AbLang2PairedConfig",
     "AutoModel": "modeling_ablang2paired.AbLang2PairedHFModel",
-    "AutoTokenizer": ["tokenizer_ablang2paired.AbLang2PairedTokenizer", "tokenizer_ablang2paired.AbLang2PairedTokenizer"]
+    "AutoTokenizer": ["tokenizer_ablang2paired.AbLang2PairedTokenizer", "tokenizer_ablang2paired.AbLang2PairedTokenizer"],
+    "AbRestore": "restoration.AbRestore",
+    "AbScores": "scores.AbScores",
+    "AbAlignment": "alignment.AbAlignment",
+    "AbEncoding": "ablang_encodings.AbEncoding",
+    "ExtraUtils": "extra_utils"
   }
 }

modeling_ablang2paired.py

@@ -21,13 +21,7 @@ except ImportError:
             "Could not find AbLang module. Please ensure ablang.py is present in the repository."
         )
 
-# Import utility files to ensure they are downloaded with the model
-# These imports ensure all utility files are included when the model is downloaded
-import restoration
-import scores
-import alignment
-import ablang_encodings
-import extra_utils
+