Update server.py

server.py

@@ -8,12 +8,13 @@ import tempfile
 
 app = FastAPI()
 
-# ✅ Fix: Set writable cache directory for Hugging Face models
-os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
-os.environ["HF_HOME"] = "/app/cache"
+# ✅ Fix: Use `/tmp` as cache directory (writable in Hugging Face Spaces)
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+os.environ["HF_HOME"] = "/tmp"
 
 # ✅ Ensure cache directory exists
-os.makedirs("/app/cache", exist_ok=True)
+if not os.path.exists("/tmp"):
+    os.makedirs("/tmp")
 
 # ✅ Load AI Model
 model_name = "deepseek-ai/DeepSeek-Coder-V2-Base"
@@ -41,7 +42,7 @@ def generate_code(request: CodeRequest):
 def test_code(request: TestRequest):
     """Runs automated testing on AI-generated code"""
     try:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".py") as temp_file:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".py", dir="/tmp") as temp_file:
             temp_file.write(request.code.encode())
             temp_file.close()