Update main.py

main.py

@@ -19,17 +19,13 @@ from transformers import pipeline
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 
-# ✅ Set Hugging Face Token for Authentication
-from huggingface_hub import login
-
 # ✅ Ensure NLTK Dependencies are Stored in a Writable Directory
 NLTK_DATA_DIR = "/tmp/nltk_data"  
 os.makedirs(NLTK_DATA_DIR, exist_ok=True)
 nltk.data.path.append(NLTK_DATA_DIR)
-try:
-    nltk.data.find("tokenizers/punkt")
-except LookupError:
-    nltk.download("punkt", download_dir=NLTK_DATA_DIR)
+
+# ✅ Fix: Download only 'punkt' (NOT 'punkt_tab')
+nltk.download("punkt", download_dir=NLTK_DATA_DIR)
 
 # ✅ Initialize FastAPI App
 app = FastAPI()