Hammad712 commited on
Commit
d8e677e
·
verified ·
1 Parent(s): f09dd6e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -8,6 +8,9 @@ from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
8
  from librosa.sequence import dtw
9
  from contextlib import asynccontextmanager
10
 
 
 
 
11
  # --- Core Class Definition ---
12
  class QuranRecitationComparer:
13
  def __init__(self, model_name="jonatasgrosman/wav2vec2-large-xlsr-53-arabic", auth_token=None):
 
8
  from librosa.sequence import dtw
9
  from contextlib import asynccontextmanager
10
 
11
+ os.environ["NUMBA_CACHE_DIR"] = "/tmp" # Ensure Numba caching works in container environments
12
+
13
+
14
  # --- Core Class Definition ---
15
  class QuranRecitationComparer:
16
  def __init__(self, model_name="jonatasgrosman/wav2vec2-large-xlsr-53-arabic", auth_token=None):