walaa2022 commited on
Commit
57061b5
·
verified ·
1 Parent(s): d4a43d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -12
app.py CHANGED
@@ -13,24 +13,22 @@ from dotenv import load_dotenv
13
  load_dotenv()
14
 
15
  # Setup logging
16
- logging.basicConfig(level=logging.INFO)
17
- logger = logging.getLogger(__name__)
18
 
19
  # Get token securely from environment variable
20
  hf_token = os.getenv('HUGGINGFACE_TOKEN')
21
- if not hf_token:
22
- logger.error("HUGGINGFACE_TOKEN not found in environment variables")
23
- raise ValueError("Please set HUGGINGFACE_TOKEN environment variable")
24
 
25
- # Login to Hugging Face
26
- try:
 
 
 
27
  login(token=hf_token)
28
- logger.info("Successfully logged in to Hugging Face")
29
- except Exception as e:
30
- logger.error(f"Error logging in to Hugging Face: {str(e)}")
31
- raise
32
 
33
-
34
  class FinancialAnalyzer:
35
  def __init__(self):
36
  """Initialize models with error handling"""
 
13
  load_dotenv()
14
 
15
  # Setup logging
16
+ huggingface-cli login
 
17
 
18
  # Get token securely from environment variable
19
  hf_token = os.getenv('HUGGINGFACE_TOKEN')
 
 
 
20
 
21
+
22
+
23
+ # Check if the token is available
24
+ if hf_token:
25
+ # Log in to Hugging Face Hub
26
  login(token=hf_token)
27
+ print("Successfully logged in to Hugging Face Hub.")
28
+ else:
29
+ print("HF_TOKEN environment variable not found. Please set it in the Space settings.")
30
+
31
 
 
32
  class FinancialAnalyzer:
33
  def __init__(self):
34
  """Initialize models with error handling"""