nexusbert commited on
Commit
4e0cc0a
·
1 Parent(s): 102ad84
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -84,6 +84,9 @@ def _get_mms():
84
  return mms_model, mms_processor
85
 
86
  hf_token = os.getenv("HF_TOKEN")
 
 
 
87
  try:
88
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
89
  logger.info("Lazy-loading MMS ASR model...")
@@ -93,8 +96,8 @@ def _get_mms():
93
  mms_model.eval()
94
  logger.info("Loaded MMS ASR model")
95
  return mms_model, mms_processor
96
- except Exception:
97
- logger.exception("Failed to load MMS ASR model")
98
  mms_model, mms_processor = None, None
99
  def _get_igbo_asr():
100
  global igbo_model, igbo_processor
 
84
  return mms_model, mms_processor
85
 
86
  hf_token = os.getenv("HF_TOKEN")
87
+ if hf_token:
88
+ hf_token = hf_token.strip()
89
+
90
  try:
91
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
92
  logger.info("Lazy-loading MMS ASR model...")
 
96
  mms_model.eval()
97
  logger.info("Loaded MMS ASR model")
98
  return mms_model, mms_processor
99
+ except Exception as e:
100
+ logger.exception(f"Failed to load MMS ASR model: {e}")
101
  mms_model, mms_processor = None, None
102
  def _get_igbo_asr():
103
  global igbo_model, igbo_processor