Update main.py

main.py

@@ -12,10 +12,11 @@ from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 
 # ✅ Ensure NLTK Dependencies are Stored in a Writable Directory
-NLTK_DATA_DIR = os.path.join(os.getcwd(), "nltk_data")
-os.makedirs(NLTK_DATA_DIR, exist_ok=True)  # Ensure the directory exists
-nltk.data.path.append(NLTK_DATA_DIR)  # Set the download path
+NLTK_DATA_DIR = "/tmp/nltk_data"  # ✅ Store in /tmp/ (writable in Hugging Face Spaces)
+os.makedirs(NLTK_DATA_DIR, exist_ok=True)
+nltk.data.path.append(NLTK_DATA_DIR)
 
+# ✅ Download punkt tokenizer
 try:
     nltk.data.find("tokenizers/punkt")
 except LookupError:
@@ -74,7 +75,7 @@ async def summarize_pdf(file: UploadFile = File(...)):
     try:
         start_time = time.time()
         pdf_content = await file.read()
-        pdf_path = "temp.pdf"
+        pdf_path = "/tmp/temp.pdf"  # ✅ Store in /tmp/
         with open(pdf_path, "wb") as f:
             f.write(pdf_content)