fair

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:
@@ -34,9 +39,9 @@ async def download_audio(
                 "preferredquality": "192",
             }
         ],
-        "outtmpl": os.path.join(tempfile.gettempdir(), "%(title)s.%(ext)s"),
+        "outtmpl": os.path.join(tempfile.gettempdir(), "%(title)s.mp3"),
     }
-    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
     )

App/Worker.py

@@ -16,7 +16,7 @@ def transcription_task(self, file_path, model_size="tiny"):
 
 
 @celery.task(name="download", bind=True)
-def downloadfile(self, url, ydl_opts):
+def downloadfile(self, url, ydl_opts, model_size="tiny"):
     # updated
     self.update_state(state="Downloading File..", meta={})
 
@@ -28,4 +28,4 @@ def downloadfile(self, url, ydl_opts):
     # updated
     self.update_state(state="Downloading complete", meta={})
 
-    return transcribe_file(state=self, file_path=audio_file, model_size="tiny")
+    return transcribe_file(state=self, file_path=audio_file, model_size=model_size)

app.py

@@ -20,6 +20,6 @@ def run_command(command):
 
 
 # Example usage
-command = "uvicorn App.app:app --host 0.0.0.0 --port 7860"
+command = "uvicorn App.app:app --host 0.0.0.0 --port 7860 & celery -A App.Worker.celery worker -c 8 --loglevel=info"
 run_command(command)
-"python3.10 -m celery -A App.Worker.celery worker -c 8 --loglevel=info"
+# "python3.10 -m celery -A App.Worker.celery worker -c 8 --loglevel=info"