Upload trained model

hanuman_loader.py

@@ -19,7 +19,7 @@ import torch
 from transformers import AutoTokenizer
 from modeling import Hanuman
 import json
-import os
+from huggingface_hub import hf_hub_download
 
 
 class HanumanModel:
@@ -40,8 +40,8 @@ class HanumanModel:
         # Download tokenizer
         tokenizer = AutoTokenizer.from_pretrained(repo_id)
 
-        # Download config
-        config_path = os.path.join(repo_id, "config.json")  # This will be downloaded by HF
+        # Download config file from HF Hub
+        config_path = hf_hub_download(repo_id, "config.json")
         with open(config_path, "r", encoding="utf-8") as f:
             cfg = json.load(f)
 
@@ -55,8 +55,8 @@ class HanumanModel:
             use_think_head=True  # Assuming your model uses this
         )
 
-        # Load weights
-        model_path = os.path.join(repo_id, "pytorch_model.bin")
+        # Download and load model weights
+        model_path = hf_hub_download(repo_id, "pytorch_model.bin")
         state = torch.load(model_path, map_location=map_location)
         model.load_state_dict(state)