Update app.py

app.py

@@ -6,12 +6,13 @@ import numpy as np
 from scipy import signal
 import os
 
-# Set up cache directory
-os.environ['TRANSFORMERS_CACHE'] = '/code/.cache'
-os.makedirs('/code/.cache', exist_ok=True)
+# Set cache directory to user's home
+cache_dir = os.path.expanduser("~/cache")
+os.environ['TRANSFORMERS_CACHE'] = cache_dir
+os.makedirs(cache_dir, 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')
+processor = WhisperProcessor.from_pretrained("openai/whisper-large-v3", cache_dir=cache_dir)
+model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v3", cache_dir=cache_dir)
 
 def process_audio(audio_path):
     waveform, sr = sf.read(audio_path)