added model support

App/Transcription/TranscriptionRoutes.py

@@ -21,6 +21,11 @@ transcription_router = APIRouter(tags=["Transcription"])
 async def download_audio(
     url: str,
     userId: int = 1,
+    model: str = Query(
+        "tiny",
+        enum=["tiny", "small", "medium", "base", "large-v2"],
+        description="Whisper model Sizes",
+    ),
 ):
     user = await User.objects.filter(id=userId).first()
     if user == None:
@@ -29,7 +34,7 @@ async def download_audio(
         "format": "bestaudio/best",
         "outtmpl": os.path.join("./", "%(title)s.%(ext)s"),
     }
-    task = downloadfile.delay(url, ydl_opts)
+    task = downloadfile.delay(url, ydl_opts, model)
     transcription_enrty = await Transcriptions.objects.create(
         task_id=task.id, user=user
     )