Update main.py

main.py

@@ -1,9 +1,14 @@
+import os
 from fastapi import FastAPI, File, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
+from huggingface_hub import login
 from NatureLM.models import NatureLM
 from NatureLM.infer import Pipeline
 import tempfile
 
+# Authenticate with HuggingFace to access gated models
+login(token=os.environ.get("HF_TOKEN"))
+
 app = FastAPI()
 
 # Allow CORS for all origins (for frontend integration)