Cartinoe5930 commited on
Commit
496b306
·
1 Parent(s): d19311f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,13 +8,13 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
8
 
9
  question_selector_map = {}
10
 
11
- model_list = ["llama2", "llama2-chat", "vicuna", "falcon", "falcon-instruct", "orca", "wizardlm"]
12
 
13
  with open("src/inference_endpoint.json", "r") as f:
14
  inference_endpoint = json.load(f)
15
 
16
- for i in range(len(model_list)):
17
- inference_endpoint[model_list[i]]["headers"]["Authorization"] += HF_TOKEN
18
 
19
  def build_question_selector_map(questions):
20
  question_selector_map = {}
 
8
 
9
  question_selector_map = {}
10
 
11
+ every_model = ["llama2", "llama2-chat", "vicuna", "falcon", "falcon-instruct", "orca", "wizardlm"]
12
 
13
  with open("src/inference_endpoint.json", "r") as f:
14
  inference_endpoint = json.load(f)
15
 
16
+ for i in range(len(every_model)):
17
+ inference_endpoint[every_model[i]]["headers"]["Authorization"] += HF_TOKEN
18
 
19
  def build_question_selector_map(questions):
20
  question_selector_map = {}