walaa2022 commited on
Commit
ceb9625
·
verified ·
1 Parent(s): 37a163f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -0
app.py CHANGED
@@ -5,11 +5,31 @@ import torch
5
  import sys
6
  import logging
7
  import io
 
 
 
 
 
8
 
9
  # Setup logging
10
  logging.basicConfig(level=logging.INFO)
11
  logger = logging.getLogger(__name__)
12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  class FinancialAnalyzer:
14
  def __init__(self):
15
  """Initialize models with error handling"""
 
5
  import sys
6
  import logging
7
  import io
8
+ from huggingface_hub import login
9
+ from dotenv import load_dotenv
10
+
11
+ # Load environment variables
12
+ load_dotenv()
13
 
14
  # Setup logging
15
  logging.basicConfig(level=logging.INFO)
16
  logger = logging.getLogger(__name__)
17
 
18
+ # Get token securely from environment variable
19
+ hf_token = os.getenv('HUGGINGFACE_TOKEN')
20
+ if not hf_token:
21
+ logger.error("HUGGINGFACE_TOKEN not found in environment variables")
22
+ raise ValueError("Please set HUGGINGFACE_TOKEN environment variable")
23
+
24
+ # Login to Hugging Face
25
+ try:
26
+ login(token=hf_token)
27
+ logger.info("Successfully logged in to Hugging Face")
28
+ except Exception as e:
29
+ logger.error(f"Error logging in to Hugging Face: {str(e)}")
30
+ raise
31
+
32
+
33
  class FinancialAnalyzer:
34
  def __init__(self):
35
  """Initialize models with error handling"""