logging
Browse files- App/Worker.py +5 -0
- App/app.py +4 -0
- app.py +1 -1
- requirements.txt +2 -1
App/Worker.py
CHANGED
@@ -17,10 +17,15 @@ def transcription_task(self, file_path, model_size="tiny"):
|
|
17 |
|
18 |
@celery.task(name="download", bind=True)
|
19 |
def downloadfile(self, url, ydl_opts):
|
|
|
20 |
self.update_state(state="Downloading File..", meta={})
|
|
|
|
|
21 |
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
22 |
info = ydl.extract_info(url, download=True)
|
23 |
audio_file = ydl.prepare_filename(info)
|
|
|
|
|
24 |
self.update_state(state="Downloading complete", meta={})
|
25 |
|
26 |
return transcribe_file(state=self, file_path=audio_file, model_size="tiny")
|
|
|
17 |
|
18 |
@celery.task(name="download", bind=True)
|
19 |
def downloadfile(self, url, ydl_opts):
|
20 |
+
# updated
|
21 |
self.update_state(state="Downloading File..", meta={})
|
22 |
+
|
23 |
+
####
|
24 |
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
25 |
info = ydl.extract_info(url, download=True)
|
26 |
audio_file = ydl.prepare_filename(info)
|
27 |
+
|
28 |
+
# updated
|
29 |
self.update_state(state="Downloading complete", meta={})
|
30 |
|
31 |
return transcribe_file(state=self, file_path=audio_file, model_size="tiny")
|
App/app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
from fastapi import FastAPI
|
2 |
from App import bot
|
|
|
3 |
from .Users.UserRoutes import user_router
|
4 |
from .modelInit import models, database
|
5 |
from .Transcription.TranscriptionRoutes import transcription_router
|
@@ -7,6 +8,9 @@ from .Streaming.StreamingRoutes import streaming_router
|
|
7 |
from .Monitor.monitorRoutes import monitor_router
|
8 |
from fastapi.middleware.cors import CORSMiddleware
|
9 |
import logging
|
|
|
|
|
|
|
10 |
|
11 |
# Configure logging
|
12 |
logging.basicConfig(
|
|
|
1 |
from fastapi import FastAPI
|
2 |
from App import bot
|
3 |
+
|
4 |
from .Users.UserRoutes import user_router
|
5 |
from .modelInit import models, database
|
6 |
from .Transcription.TranscriptionRoutes import transcription_router
|
|
|
8 |
from .Monitor.monitorRoutes import monitor_router
|
9 |
from fastapi.middleware.cors import CORSMiddleware
|
10 |
import logging
|
11 |
+
import transformers
|
12 |
+
|
13 |
+
transformers.logging.set_verbosity_info()
|
14 |
|
15 |
# Configure logging
|
16 |
logging.basicConfig(
|
app.py
CHANGED
@@ -20,5 +20,5 @@ def run_command(command):
|
|
20 |
|
21 |
|
22 |
# Example usage
|
23 |
-
command = "uvicorn App.app:app --host 0.0.0.0 --port 7860 & celery -A App.Worker.celery worker --loglevel=info"
|
24 |
run_command(command)
|
|
|
20 |
|
21 |
|
22 |
# Example usage
|
23 |
+
command = "uvicorn App.app:app --host 0.0.0.0 --port 7860 & celery -A App.Worker.celery worker -c 8 --loglevel=info"
|
24 |
run_command(command)
|
requirements.txt
CHANGED
@@ -5,7 +5,8 @@ fastapi==0.92.0
|
|
5 |
Flask==2.2.2
|
6 |
greenlet==2.0.2
|
7 |
itsdangerous==2.1.2
|
8 |
-
orm==0.3.
|
|
|
9 |
faster-whisper
|
10 |
aiofiles
|
11 |
psycopg2-binary==2.9.5
|
|
|
5 |
Flask==2.2.2
|
6 |
greenlet==2.0.2
|
7 |
itsdangerous==2.1.2
|
8 |
+
orm==0.3.
|
9 |
+
transformers
|
10 |
faster-whisper
|
11 |
aiofiles
|
12 |
psycopg2-binary==2.9.5
|