Update main.py

main.py

@@ -1,3 +1,11 @@
+import os
+
+# Set the cache directory to a writable directory
+cache_dir = "/tmp/huggingface_cache"  # Contoh direktori yang bisa ditulis
+if not os.path.exists(cache_dir):
+    os.makedirs(cache_dir, exist_ok=True)
+os.environ["TRANSFORMERS_CACHE"] = cache_dir
+
 import torch
 from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
 from datasets import load_dataset