UniqAI/openedai-speech
1
0
Fork 0
mirror of https://github.com/matatonic/openedai-speech synced 2025-06-26 18:16:32 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

minor response type update

Browse Source
This commit is contained in:
matatonic 2023-12-01 15:41:01 -05:00
parent 98c102d0ef
commit 3fe07873f7
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.py
View File

@ -7,7 +7,7 @@ import subprocess
import tempfile
import yaml
from fastapi import FastAPI
from fastapi.responses import StreamingResponse, Response
from fastapi.responses import StreamingResponse, PlainTextResponse
from fastapi.middleware.cors import CORSMiddleware
import uvicorn
from pydantic import BaseModel
@ -57,11 +57,11 @@ app.add_middleware(
allow_headers=["*"]
)
@app.get("/")
@app.head("/")
@app.options("/")
async def root() -> Response:
return Response(status_code=200, content="", media_type="text/plain")
@app.get("/", response_class=PlainTextResponse)
@app.head("/", response_class=PlainTextResponse)
@app.options("/", response_class=PlainTextResponse)
async def root():
return PlainTextResponse(content="")
class GenerateSpeechRequest(BaseModel):
model: str = "tts-1" # or "tts-1-hd"
@ -70,7 +70,7 @@ class GenerateSpeechRequest(BaseModel):
response_format: str = "mp3" # mp3, opus, aac, flac
speed: float = 1.0 # 0.25 - 4.0
@app.post("/v1/audio/speech")
@app.post("/v1/audio/speech", response_class=StreamingResponse)
async def generate_speech(request: GenerateSpeechRequest):
global xtts, args
input_text = preprocess(request.input)