Update app.py

app.py

@@ -5,11 +5,31 @@ import torch
 import sys
 import logging
 import io
+from huggingface_hub import login
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Get token securely from environment variable
+hf_token = os.getenv('HUGGINGFACE_TOKEN')
+if not hf_token:
+    logger.error("HUGGINGFACE_TOKEN not found in environment variables")
+    raise ValueError("Please set HUGGINGFACE_TOKEN environment variable")
+
+# Login to Hugging Face
+try:
+    login(token=hf_token)
+    logger.info("Successfully logged in to Hugging Face")
+except Exception as e:
+    logger.error(f"Error logging in to Hugging Face: {str(e)}")
+    raise
+
+    
 class FinancialAnalyzer:
     def __init__(self):
         """Initialize models with error handling"""