Update app.py

app.py

@@ -4,9 +4,14 @@ from transformers import WhisperProcessor, WhisperForConditionalGeneration
 import soundfile as sf
 import numpy as np
 from scipy import signal
+import os
 
-processor = WhisperProcessor.from_pretrained("openai/whisper-large-v3")
-model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v3")
+# Set up cache directory
+os.environ['TRANSFORMERS_CACHE'] = '/code/.cache'
+os.makedirs('/code/.cache', exist_ok=True)
+
+processor = WhisperProcessor.from_pretrained("openai/whisper-large-v3", cache_dir='/code/.cache')
+model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v3", cache_dir='/code/.cache')
 
 def process_audio(audio_path):
     waveform, sr = sf.read(audio_path)