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",
|
|
|
|
|
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
|