Add application file

Dockerfile

@@ -3,13 +3,6 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 COPY ./s2smodels.py /code/
-COPY ./macros.py /code/
-COPY ./utils/ . /code/ 
-COPY ./modules/ . /code/ 
-COPY ./models/ . /code/ 
-COPY ./data/ . /code/
-COPY ./prompts/ . /code/ 
-COPY ./customs/ . /code/ 
 COPY ./main.py /code/ 
 
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]

main.py

@@ -7,9 +7,9 @@ import os
 from fastapi import FastAPI, Response
 import torch
 from fastapi.responses import JSONResponse
-from utils.prompt_making import make_prompt
+from VELL-X.utils.prompt_making import make_prompt
 from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
-from utils.generation import SAMPLE_RATE, generate_audio, preload_models
+from VELL-X.utils.generation import SAMPLE_RATE, generate_audio, preload_models
 from io import BytesIO
 from pyannote.audio import Pipeline
 import soundfile as sf