Initial upload: HS Code Classifier (English, 6-digit)

inference.py

@@ -9,7 +9,7 @@ import math
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-MODEL_DIR = 'models_4090_eng_6digit'
+MODEL_DIR = 'model'
 FULL_MODEL_PATH = os.path.join(MODEL_DIR, 'cascaded_best.pt')
 CONFIG_PATH = os.path.join(MODEL_DIR, 'model_config.json')
 TOKENIZER_PATH = os.path.join(MODEL_DIR, 'tokenizer')