sachin commited on
Commit
998dfd0
·
1 Parent(s): 2ef8210
Files changed (1) hide show
  1. 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
- self.repo_id,
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)