Spaces:
Running
Running
Update apis/chat_api.py
Browse files- apis/chat_api.py +6 -3
apis/chat_api.py
CHANGED
|
@@ -17,6 +17,7 @@ import re
|
|
| 17 |
import requests
|
| 18 |
from utils.enver import enver
|
| 19 |
import shutil
|
|
|
|
| 20 |
|
| 21 |
|
| 22 |
from fastapi import FastAPI, Response, File, UploadFile, Form
|
|
@@ -31,7 +32,8 @@ from googletrans import Translator
|
|
| 31 |
from io import BytesIO
|
| 32 |
from gtts import gTTS
|
| 33 |
from fastapi.middleware.cors import CORSMiddleware
|
| 34 |
-
|
|
|
|
| 35 |
|
| 36 |
class ChatAPIApp:
|
| 37 |
def __init__(self):
|
|
@@ -309,7 +311,7 @@ app.add_middleware(
|
|
| 309 |
)
|
| 310 |
@app.post("/transcribe")
|
| 311 |
async def whisper_transcribe(
|
| 312 |
-
audio_file:
|
| 313 |
language: str = Form(),
|
| 314 |
model: str = Form(),
|
| 315 |
):
|
|
@@ -346,7 +348,8 @@ async def whisper_transcribe(
|
|
| 346 |
)
|
| 347 |
time_start = time.time()
|
| 348 |
pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=language, task="transcribe")
|
| 349 |
-
|
|
|
|
| 350 |
time_end = time.time()
|
| 351 |
item_response["status"] = 200
|
| 352 |
item_response["result"] = text
|
|
|
|
| 17 |
import requests
|
| 18 |
from utils.enver import enver
|
| 19 |
import shutil
|
| 20 |
+
import tempfile
|
| 21 |
|
| 22 |
|
| 23 |
from fastapi import FastAPI, Response, File, UploadFile, Form
|
|
|
|
| 32 |
from io import BytesIO
|
| 33 |
from gtts import gTTS
|
| 34 |
from fastapi.middleware.cors import CORSMiddleware
|
| 35 |
+
from pathlib import Path
|
| 36 |
+
from tempfile import NamedTemporaryFile
|
| 37 |
|
| 38 |
class ChatAPIApp:
|
| 39 |
def __init__(self):
|
|
|
|
| 311 |
)
|
| 312 |
@app.post("/transcribe")
|
| 313 |
async def whisper_transcribe(
|
| 314 |
+
audio_file: UploadFile = File(description="Audio file for transcribe"),
|
| 315 |
language: str = Form(),
|
| 316 |
model: str = Form(),
|
| 317 |
):
|
|
|
|
| 348 |
)
|
| 349 |
time_start = time.time()
|
| 350 |
pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=language, task="transcribe")
|
| 351 |
+
file_data = await audio_file.read()
|
| 352 |
+
text = pipe(file_data)["text"]
|
| 353 |
time_end = time.time()
|
| 354 |
item_response["status"] = 200
|
| 355 |
item_response["result"] = text
|