Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"""
|