3v324v23 commited on
Commit
8742a91
1 Parent(s): 5cc4420

Add application file

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -7,9 +7,9 @@ import os
7
  from fastapi import FastAPI, Response
8
  import torch
9
  from fastapi.responses import JSONResponse
10
- from VALLE-X.utils.prompt_making import make_prompt
11
  from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
12
- from VALLE-X.utils.generation import SAMPLE_RATE, generate_audio, preload_models
13
  from io import BytesIO
14
  from pyannote.audio import Pipeline
15
  import soundfile as sf
 
7
  from fastapi import FastAPI, Response
8
  import torch
9
  from fastapi.responses import JSONResponse
10
+ from utils.prompt_making import make_prompt
11
  from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
12
+ from utils.generation import SAMPLE_RATE, generate_audio, preload_models
13
  from io import BytesIO
14
  from pyannote.audio import Pipeline
15
  import soundfile as sf