Update main.py (#4)

- Update main.py (1df55422db7d0a17428edcaf3664f1635639a114)

main.py

@@ -10,13 +10,16 @@ import torch
 import soundfile as sf
 import numpy as np
 from kokoro import KModel, KPipeline
+import os
+
+os.environ['HUGGINGFACE_HUB_CACHE'] = '/app/models'
 
 app = FastAPI()
 
 CUDA_AVAILABLE = torch.cuda.is_available()
 models = {gpu: KModel().to('cuda' if gpu else 'cpu').eval() for gpu in [False] + ([True] if CUDA_AVAILABLE else [])}
 pipelines = {lang_code: KPipeline(lang_code=lang_code, model=False) for lang_code in ['a', 'p', 'e']}
-
+# ... (restante do código inalterado)
 CHOICES = {
     '🇺🇸 🚺 Heart ❤️': 'af_heart',
     '🇺🇸 🚺 Alloy': 'af_alloy',