last trt to download the model at build time using the scripts

download_model.py

@@ -25,17 +25,34 @@ def download_model():
         # Create local directory if it doesn't exist
         os.makedirs(LOCAL_MODEL_PATH, exist_ok=True)
         
-        # Use huggingface_hub to download the model
-        from huggingface_hub import snapshot_download
+        # Use huggingface_hub to download the model files
+        from huggingface_hub import hf_hub_download, list_repo_files
         
-        # Download the int4 model files using subfolder
-        snapshot_download(
-            repo_id=MAIN_MODEL_ID,
-            subfolder="int4",  # Use the int4 subfolder
-            local_dir=LOCAL_MODEL_PATH,
-            local_dir_use_symlinks=False,
-            ignore_patterns=["*.md", "*.txt", "*.git*", "*.ipynb", "*.py"]
-        )
+        # List files in the int4 subfolder
+        files = list_repo_files(MAIN_MODEL_ID, subfolder="int4")
+        
+        # Download each required file
+        required_files = [
+            "config.json",
+            "pytorch_model.bin", 
+            "tokenizer.json",
+            "tokenizer_config.json",
+            "special_tokens_map.json",
+            "generation_config.json"
+        ]
+        
+        for file_name in required_files:
+            if file_name in files:
+                logger.info(f"Downloading {file_name}...")
+                hf_hub_download(
+                    repo_id=MAIN_MODEL_ID,
+                    filename=f"int4/{file_name}",
+                    local_dir=LOCAL_MODEL_PATH,
+                    local_dir_use_symlinks=False
+                )
+                logger.info(f"Downloaded {file_name}")
+            else:
+                logger.warning(f"File {file_name} not found in int4 subfolder")
         
         logger.info(f"Model downloaded successfully to {LOCAL_MODEL_PATH}")
         return True

test_download.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+"""
+Simple test script to verify download functionality for Hugging Face Spaces
+"""
+
+import os
+import sys
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+def test_download():
+    """Test the download functionality"""
+    try:
+        logger.info("Testing download functionality...")
+        
+        # Import and run the download script
+        from download_model import main as download_main
+        
+        success = download_main()
+        logger.info(f"Download test result: {'PASS' if success else 'FAIL'}")
+        
+        return success
+        
+    except Exception as e:
+        logger.error(f"Error testing download: {e}")
+        return False
+
+def main():
+    """Main test function"""
+    logger.info("Starting download test...")
+    
+    success = test_download()
+    
+    if success:
+        logger.info("✅ Download test passed!")
+    else:
+        logger.error("❌ Download test failed!")
+    
+    return success
+
+if __name__ == "__main__":
+    success = main()
+    sys.exit(0 if success else 1)