| import os |
| import sys |
| from huggingface_hub import hf_hub_download |
|
|
| def download_model(): |
| """Download the required model for the application.""" |
| try: |
| print("Creating models directory...") |
| os.makedirs('models', exist_ok=True) |
| |
| print("Downloading model from Hugging Face...") |
| hf_hub_download( |
| repo_id='bartowski/SmolLM2-360M-Instruct-GGUF', |
| filename='SmolLM2-360M-Instruct-Q4_K_M.gguf', |
| local_dir='models', |
| local_dir_use_symlinks=False |
| ) |
| |
| print("Model downloaded successfully!") |
| return True |
| |
| except Exception as e: |
| print(f"Error downloading model: {e}", file=sys.stderr) |
| return False |
|
|
| if __name__ == "__main__": |
| success = download_model() |
| sys.exit(0 if success else 1) |
|
|
