Cartinoe5930 commited on
Commit
2887740
1 Parent(s): b074ed1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -78,10 +78,7 @@ 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
- 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
 
 
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].lower()
 
 
 
82
 
83
  model_response = Inference(model_list, question, API_KEY, cot, hf_token)
84