up

chatbot.py

@@ -49,14 +49,19 @@ def setup_cache_dirs():
     is_spaces = os.environ.get('SPACE_ID') is not None
     
     if is_spaces:
-        # Use /tmp for Hugging Face Spaces
+        # Use /tmp for Hugging Face Spaces with proper permissions
         cache_dir = '/tmp/huggingface'
         os.environ.update({
             'TRANSFORMERS_CACHE': cache_dir,
             'HF_HOME': cache_dir,
             'TOKENIZERS_PARALLELISM': 'false',
             'TRANSFORMERS_VERBOSITY': 'error',
-            'BITSANDBYTES_NOWELCOME': '1'
+            'BITSANDBYTES_NOWELCOME': '1',
+            'HF_DATASETS_CACHE': cache_dir,
+            'HF_METRICS_CACHE': cache_dir,
+            'HF_MODULES_CACHE': cache_dir,
+            'HUGGING_FACE_HUB_TOKEN': os.environ.get('HF_TOKEN', ''),
+            'HF_TOKEN': os.environ.get('HF_TOKEN', '')
         })
     else:
         # Use default cache for local development
@@ -67,8 +72,12 @@ def setup_cache_dirs():
             'BITSANDBYTES_NOWELCOME': '1'
         })
     
-    # Create cache directory if it doesn't exist
+    # Create cache directory with proper permissions
     os.makedirs(cache_dir, exist_ok=True)
+    if is_spaces:
+        # Ensure directory is writable
+        os.chmod(cache_dir, 0o777)
+    
     return cache_dir
 
 # Set up cache directories