SumakaClone / backend /scripts /bootstrap_models.py
raghava0450's picture
Deploy SumakaClone Space configuration
92d8b0d verified
Raw
History Blame Contribute Delete
571 Bytes
from __future__ import annotations
import asyncio
from app.configs.settings import get_settings
from app.services.huggingface_service import HuggingFaceService
from app.services.model_service import ModelService
async def main() -> None:
settings = get_settings()
model_service = ModelService(settings)
huggingface = HuggingFaceService(settings)
for model in model_service.list_models():
path = await huggingface.ensure_model_available(model)
print(f"{model.display_name}: {path}")
if __name__ == "__main__":
asyncio.run(main())