4lli39421 commited on
Commit
5455493
·
verified ·
1 Parent(s): 98ae1f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -5,11 +5,13 @@ import os
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  from huggingface_hub import login, HfApi
7
 
8
- # Optional: Login if you want access to gated/private models
9
- HF_TOKEN = os.getenv("HF_TOKEN", None)
10
  if HF_TOKEN:
 
11
  login(HF_TOKEN)
12
 
 
13
  # Define model map with access type
14
  model_map = {
15
  "FinGPT": {"id": "AI4Finance/FinGPT", "local": True},
 
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  from huggingface_hub import login, HfApi
7
 
8
+ HF_TOKEN = os.getenv("Allie", None)
9
+
10
  if HF_TOKEN:
11
+ from huggingface_hub import login
12
  login(HF_TOKEN)
13
 
14
+
15
  # Define model map with access type
16
  model_map = {
17
  "FinGPT": {"id": "AI4Finance/FinGPT", "local": True},