Cartinoe5930 commited on
Commit
af4b3b5
·
1 Parent(s): 54536dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -78,7 +78,10 @@ def inference(model_list, question, API_KEY, cot, hf_token=HF_TOKEN):
78
  raise gr.Error("Please choose just '3' models! Neither more nor less!")
79
 
80
  for i in range(len(model_list)):
81
- model_list[i] = model_list[i][:-1].lower()
 
 
 
82
 
83
  model_response = Inference(model_list, question, API_KEY, cot, hf_token)
84
 
 
78
  raise gr.Error("Please choose just '3' models! Neither more nor less!")
79
 
80
  for i in range(len(model_list)):
81
+ if "llama" in model_list[I].lower():
82
+ model_list[i] = model_list[i][:-2].lower()
83
+ else:
84
+ model_list[i] = model_list[i][:-1].lower()
85
 
86
  model_response = Inference(model_list, question, API_KEY, cot, hf_token)
87