Spaces:
Sleeping
Sleeping
Update Models/Utils.py
Browse files- 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 |
|