DPWXY15 commited on
Commit
e532bec
·
verified ·
1 Parent(s): 042042c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from fastapi import FastAPI, Request, UploadFile, File
2
  import openai
3
  import os
 
4
 
5
  openai.api_key = os.environ["OPENAI_API_KEY"]
6
 
@@ -18,9 +19,12 @@ async def chat(req: Request):
18
 
19
  @app.post("/transcribe")
20
  async def transcribe(file: UploadFile = File(...)):
21
- audio_data = await file.read()
22
- response = transcript = openai.audio.transcriptions.create(
23
- model="whisper-1",
24
- file=audio_data
25
- )
 
 
 
26
  return {"text": response.text}
 
1
  from fastapi import FastAPI, Request, UploadFile, File
2
  import openai
3
  import os
4
+ from io import BytesIO
5
 
6
  openai.api_key = os.environ["OPENAI_API_KEY"]
7
 
 
19
 
20
  @app.post("/transcribe")
21
  async def transcribe(file: UploadFile = File(...)):
22
+ audio_bytes = await file.read()
23
+ audio_file = BytesIO(audio_bytes)
24
+ audio_file.name = file.filename # Required so OpenAI knows the format
25
+
26
+ response = openai.audio.transcriptions.create(
27
+ model="whisper-1",
28
+ file=audio_file
29
+ )
30
  return {"text": response.text}