Spaces:
Sleeping
Sleeping
AshDavid12
commited on
Commit
·
01fe41f
1
Parent(s):
d632f1f
trying to build-tempdir
Browse files- infer.py +4 -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 |
-
|
42 |
-
|
|
|
|
|
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=
|
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:
|