taha454 commited on
Commit
70bc77d
·
verified ·
1 Parent(s): 39751d8

Update Models/Utils.py

Browse files
Files changed (1) hide show
  1. Models/Utils.py +2 -1
Models/Utils.py CHANGED
@@ -9,8 +9,9 @@ message_system = lambda x: ({"role": "system", "content": x})
9
  def get_specific_model(model_name):
10
  if model_name in ["gemini-1.5-flash","gemini"]:
11
  return Gemini()
12
- elif model_name in ["openai/gpt-oss-120b","gpt 120"]:
13
  return GPT("openai/gpt-oss-20b")
 
14
  else :
15
  return GPT("openai/gpt-oss-120b")
16
 
 
9
  def get_specific_model(model_name):
10
  if model_name in ["gemini-1.5-flash","gemini"]:
11
  return Gemini()
12
+ elif model_name in ["openai/gpt-oss-120b","gpt 120","gbt20"]:
13
  return GPT("openai/gpt-oss-20b")
14
+
15
  else :
16
  return GPT("openai/gpt-oss-120b")
17