aryo100 commited on
Commit
c02702b
·
1 Parent(s): 28b5a9e

update api

Browse files
Files changed (1) hide show
  1. api.py +5 -3
api.py CHANGED
@@ -7,10 +7,11 @@ import fairseq
7
  from fastapi import FastAPI, HTTPException
8
  from fastapi.responses import FileResponse
9
  from pydantic import BaseModel
10
- from modules import models
11
  from uuid import uuid4
12
  import requests
13
  from modules.core import preload
 
14
 
15
  app = FastAPI()
16
 
@@ -113,8 +114,9 @@ async def tts_api(req: TTSRequest):
113
 
114
  # 2. Voice Conversion
115
  try:
116
- models.load_model(ckpt_path)
117
- vc = models.vc_model
 
118
 
119
  if vc is None:
120
  raise Exception("Failed to load model")
 
7
  from fastapi import FastAPI, HTTPException
8
  from fastapi.responses import FileResponse
9
  from pydantic import BaseModel
10
+ # from modules import models
11
  from uuid import uuid4
12
  import requests
13
  from modules.core import preload
14
+ from modules.models import load_model
15
 
16
  app = FastAPI()
17
 
 
114
 
115
  # 2. Voice Conversion
116
  try:
117
+ # models.load_model(ckpt_path)
118
+ # vc = models.vc_model
119
+ vc = load_model(ckpt_path, config_json="configs/48k-768.json")
120
 
121
  if vc is None:
122
  raise Exception("Failed to load model")