uzi007 commited on
Commit
5b09a75
1 Parent(s): 8ac0fa3

Removing Models Loading

Browse files
Files changed (2) hide show
  1. api_functions.py +4 -4
  2. audiobook.py +1 -1
api_functions.py CHANGED
@@ -9,10 +9,10 @@ from languages import CODE2LANG
9
  from s3_handler import S3Handler
10
 
11
  from media_download import YoutubeDownloader
12
- from transcription import StableWhisper
13
- from translation import Translation
14
- from summarizer import Extract_Summary, AudioBookNarration
15
- from audiobook import AudioBook
16
 
17
  from celery import Celery
18
 
 
9
  from s3_handler import S3Handler
10
 
11
  from media_download import YoutubeDownloader
12
+ # from transcription import StableWhisper
13
+ # from translation import Translation
14
+ # from summarizer import Extract_Summary, AudioBookNarration
15
+ # from audiobook import AudioBook
16
 
17
  from celery import Celery
18
 
audiobook.py CHANGED
@@ -13,7 +13,7 @@ import soundfile as sf
13
  os.environ["CUDA_VISIBLE_DEVICES"] = "0"
14
 
15
  # Loads the model, should be run one time
16
- preload_models()
17
 
18
  class AudioBook:
19
  def __init__(self, output_folder="output"):
 
13
  os.environ["CUDA_VISIBLE_DEVICES"] = "0"
14
 
15
  # Loads the model, should be run one time
16
+ # preload_models()
17
 
18
  class AudioBook:
19
  def __init__(self, output_folder="output"):