sachin
commited on
Commit
·
998dfd0
1
Parent(s):
2ef8210
download
Browse files- src/server/main.py +2 -1
src/server/main.py
CHANGED
@@ -269,9 +269,10 @@ class TTSManager:
|
|
269 |
async def load(self):
|
270 |
if not self.model:
|
271 |
logger.info("Loading TTS model IndicF5 asynchronously...")
|
|
|
272 |
self.model = await asyncio.to_thread(
|
273 |
AutoModel.from_pretrained,
|
274 |
-
|
275 |
trust_remote_code=True
|
276 |
)
|
277 |
self.model = self.model.to(self.device_type)
|
|
|
269 |
async def load(self):
|
270 |
if not self.model:
|
271 |
logger.info("Loading TTS model IndicF5 asynchronously...")
|
272 |
+
local_path = "/app/models/tts_model"
|
273 |
self.model = await asyncio.to_thread(
|
274 |
AutoModel.from_pretrained,
|
275 |
+
local_path,
|
276 |
trust_remote_code=True
|
277 |
)
|
278 |
self.model = self.model.to(self.device_type)
|