Update main.py

main.py

@@ -11,7 +11,10 @@ from huggingface_hub import login
 
 app = FastAPI()
 
-# Log in to Hugging Face using environment variable
+# Set HF_HOME to a writable directory
+os.environ["HF_HOME"] = "/app/.cache"
+
+# Log in to Hugging Face
 hf_token = os.getenv("HF_TOKEN")
 if not hf_token:
     raise ValueError("HF_TOKEN environment variable not set")
@@ -20,7 +23,7 @@ login(hf_token)
 # Load Dataset and Prepare Knowledge Base
 ds = load_dataset("codeparrot/apps", "all", split="train")
 os.makedirs("knowledge_base", exist_ok=True)
-for i, example in enumerate(ds.select(range(100))):  # Reduced to 100 for free tier
+for i, example in enumerate(ds.select(range(100))):
     solution = example['solutions'][0] if example['solutions'] else "No solution available"
     with open(f"knowledge_base/doc_{i}.txt", "w", encoding="utf-8") as f:
         f.write(f"### Problem\n{example['question']}\n\n### Solution\n{solution}")