AshDavid12 commited on
Commit
01fe41f
·
1 Parent(s): d632f1f

trying to build-tempdir

Browse files
Files changed (2) hide show
  1. infer.py +4 -2
  2. whisper_online.py +2 -3
infer.py CHANGED
@@ -38,8 +38,10 @@ logging.info(f"Selected model name: {model_name}")
38
  try:
39
  lan = 'he'
40
  logging.info(f"Attempting to initialize FasterWhisperASR with device: {device}")
41
- #cache_dir = os.environ.get('XDG_CACHE_HOME', tempfile.gettempdir())
42
- model = whisper_online.FasterWhisperASR(lan=lan, modelsize=model_name, cache_dir=None, model_dir=None)
 
 
43
  logging.info("FasterWhisperASR model initialized successfully.")
44
  except Exception as e:
45
  logging.error(f"Falied to inilialize faster whisper model {e}")
 
38
  try:
39
  lan = 'he'
40
  logging.info(f"Attempting to initialize FasterWhisperASR with device: {device}")
41
+ logging.info(f"Cache directory before: {tempfile.gettempdir()}") # Log the temp directory
42
+ cache_dir = os.environ.get('XDG_CACHE_HOME', tempfile.gettempdir())
43
+ logging.info(f"Cache directory after: {tempfile.gettempdir()}") # Log the temp directory
44
+ model = whisper_online.FasterWhisperASR(lan=lan, modelsize=model_name, cache_dir=cache_dir, model_dir=None)
45
  logging.info("FasterWhisperASR model initialized successfully.")
46
  except Exception as e:
47
  logging.error(f"Falied to inilialize faster whisper model {e}")
whisper_online.py CHANGED
@@ -105,7 +105,7 @@ class FasterWhisperASR(ASRBase):
105
 
106
  sep = ""
107
 
108
- def load_model(self, modelsize=None, cache_dir="/tmp", model_dir=None):
109
  from faster_whisper import WhisperModel
110
  # logging.getLogger("faster_whisper").setLevel(logger.level)
111
 
@@ -123,8 +123,7 @@ class FasterWhisperASR(ASRBase):
123
 
124
  try:
125
  logging.info(f"Loading WhisperModel on device: ")
126
- logging.info(f"Cache directory: {tempfile.gettempdir()}") # Log the temp directory
127
- cache_dir = os.getenv('XDG_CACHE_HOME', tempfile.gettempdir())
128
  model = WhisperModel(model_size_or_path, device="cuda", compute_type="float16", download_root=cache_dir)
129
  logging.info("Model loaded successfully.")
130
  except Exception as e:
 
105
 
106
  sep = ""
107
 
108
+ def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
109
  from faster_whisper import WhisperModel
110
  # logging.getLogger("faster_whisper").setLevel(logger.level)
111
 
 
123
 
124
  try:
125
  logging.info(f"Loading WhisperModel on device: ")
126
+ logging.info(f"Cache directory in online: {tempfile.gettempdir()}") # Log the temp directory
 
127
  model = WhisperModel(model_size_or_path, device="cuda", compute_type="float16", download_root=cache_dir)
128
  logging.info("Model loaded successfully.")
129
  except Exception as e: