Update app.py

app.py

@@ -9,7 +9,7 @@ import os
 model_repo = "elapt1c/hrom-testing"
 
 # 1. Import trainer module components
-trainer_file = hf_hub_download(repo_id=model_repo, filename="HROM-V1.5_Trainer.py")
+trainer_file = hf_hub_download(repo_id=model_repo, filename="HROM_Trainer.py")
 spec = importlib.util.spec_from_file_location("HROM_Trainer", trainer_file)
 trainer_module = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(trainer_module)
@@ -18,11 +18,11 @@ CONFIG = trainer_module.CONFIG
 SafetyManager = trainer_module.SafetyManager
 
 # 2. Load tokenizer
-tokenizer_file = hf_hub_download(repo_id=model_repo, filename="tokenizer/hrom_tokenizer.json")
+tokenizer_file = hf_hub_download(repo_id=model_repo, filename="hrom_tokenizer.json")
 tokenizer = Tokenizer.from_file(tokenizer_file)
 
 # 3. Load model checkpoint
-checkpoint_file = hf_hub_download(repo_id=model_repo, filename="HROM-V1.5_Trained-Model/HROM-V1.5.pt")
+checkpoint_file = hf_hub_download(repo_id=model_repo, filename="hrom.pt")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 def load_model():