Mbonea commited on
Commit
7306982
1 Parent(s): fec71d6
Files changed (4) hide show
  1. App/Worker.py +5 -0
  2. App/app.py +4 -0
  3. app.py +1 -1
  4. 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.1
 
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