Update app.py

app.py

@@ -5,12 +5,18 @@ from datetime import datetime
 import torch
 from transformers import PegasusForConditionalGeneration, PegasusTokenizer
 import time
+#from dotenv import load_dotenv
+import os
+
+#load_dotenv()
 
 app = FastAPI()
 
+API_KEY = os.getenv("API_KEY")
+
 # Configuration
 API_KEYS = {
-    "bdLFqk4IcYmRE2ONZeCts4DWrqkpqQxW": "user1"  # In production, use a secure database
+    API_KEY : "user1"  # In production, use a secure database
 }
 
 # Initialize model and tokenizer with smaller model for Spaces