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

Add application file

Browse files
Files changed (2) hide show
  1. main.py +2 -2
  2. requirements.txt +1 -0
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 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
 
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
requirements.txt CHANGED
@@ -34,3 +34,4 @@ fastapi-cors
34
  sqlalchemy
35
  sqlalchemy.orm
36
 
 
 
34
  sqlalchemy
35
  sqlalchemy.orm
36
 
37
+ git+https://github.com/Plachtaa/VALL-E-X.git