Fix initial bugs

Dockerfile

@@ -19,12 +19,19 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY . .
 
 # Create necessary directories
-RUN mkdir -p csgo/spawn config checkpoints cache
+RUN mkdir -p csgo/spawn config checkpoints cache && \
+    mkdir -p /tmp/torch /tmp/huggingface /tmp/transformers
 
 # Set environment variables
 ENV PYTHONPATH=/app/src:/app
 ENV CUDA_VISIBLE_DEVICES=""
-ENV OMP_NUM_THREADS=4
+ENV OMP_NUM_THREADS=2
+ENV MKL_NUM_THREADS=2
+
+# Set cache directories to writable locations
+ENV TORCH_HOME=/tmp/torch
+ENV HF_HOME=/tmp/huggingface
+ENV TRANSFORMERS_CACHE=/tmp/transformers
 
 # Expose port
 EXPOSE 7860

app.py

@@ -9,6 +9,7 @@ import io
 import json
 import logging
 import os
+import tempfile
 from pathlib import Path
 from typing import Dict, List, Optional, Set
 
@@ -51,7 +52,21 @@ app = FastAPI(title="Diamond CSGO AI Player")
 os.environ.setdefault("SDL_VIDEODRIVER", "dummy")
 os.environ.setdefault("SDL_AUDIODRIVER", "dummy")
 os.environ.setdefault("PYGAME_HIDE_SUPPORT_PROMPT", "1")
-os.environ.setdefault("OMP_NUM_THREADS", "1")
+
+# Fix OMP_NUM_THREADS for HF Spaces (must be positive integer)
+if "OMP_NUM_THREADS" not in os.environ or not os.environ["OMP_NUM_THREADS"].isdigit():
+    os.environ["OMP_NUM_THREADS"] = "2"
+
+# Set up proper cache directories for HF Spaces
+temp_dir = tempfile.gettempdir()
+os.environ.setdefault("TORCH_HOME", os.path.join(temp_dir, "torch"))
+os.environ.setdefault("HF_HOME", os.path.join(temp_dir, "huggingface"))
+os.environ.setdefault("TRANSFORMERS_CACHE", os.path.join(temp_dir, "transformers"))
+
+# Create cache directories
+for cache_var in ["TORCH_HOME", "HF_HOME", "TRANSFORMERS_CACHE"]:
+    cache_path = os.environ[cache_var]
+    os.makedirs(cache_path, exist_ok=True)
 connected_clients: Set[WebSocket] = set()
 
 class WebKeyMap:
@@ -117,9 +132,20 @@ class WebGameEngine:
                 
                 model_url = "https://huggingface.co/Etadingrui/diamond-1B/resolve/main/agent_epoch_00003.pt"
                 
-                # Use torch.hub to download and load state dict
+                # Use torch.hub to download and load state dict with custom cache dir
                 logger.info(f"Loading state dict from {model_url}")
-                state_dict = torch.hub.load_state_dict_from_url(model_url, map_location=device)
+                
+                # Set custom cache directory that we have write permissions for
+                cache_dir = os.path.join(tempfile.gettempdir(), "torch_cache")
+                os.makedirs(cache_dir, exist_ok=True)
+                
+                # Use torch.hub with custom cache directory
+                state_dict = torch.hub.load_state_dict_from_url(
+                    model_url, 
+                    map_location=device,
+                    model_dir=cache_dir,
+                    check_hash=False  # Skip hash check for faster loading
+                )
                 
                 self.download_progress = 60
                 self.loading_status = "Loading model weights into agent..."