Mbonea commited on
Commit
f61cd1c
1 Parent(s): fe76727

added model choices

Browse files
App/Transcription/TranscriptionRoutes.py CHANGED
@@ -17,7 +17,9 @@ async def create_file(
17
  file: UploadFile,
18
  userId: int = 1,
19
  model: str = Query(
20
- "tiny", enum=["tiny", "small", "medium"], description="Model Sizes"
 
 
21
  ),
22
  ):
23
  # Read the file contents
 
17
  file: UploadFile,
18
  userId: int = 1,
19
  model: str = Query(
20
+ "tiny",
21
+ enum=["tiny", "small", "medium", "base", "large"],
22
+ description="Whisper model Sizes",
23
  ),
24
  ):
25
  # Read the file contents