Update crs_arena/utils.py
Browse files- crs_arena/utils.py +0 -28
crs_arena/utils.py
CHANGED
@@ -59,34 +59,6 @@ def get_crs_model(model_name: str, model_config_file: str) -> CRSModel:
|
|
59 |
|
60 |
return CRSModel(name, **model_args), ttl=timedelta(days=3)
|
61 |
)
|
62 |
-
def get_crs_model(model_name: str, model_config_file: str) -> CRSModel:
|
63 |
-
"""Returns a CRS model.
|
64 |
-
|
65 |
-
Args:
|
66 |
-
model_name: Model name.
|
67 |
-
model_config_file: Model configuration file.
|
68 |
-
|
69 |
-
Raises:
|
70 |
-
FileNotFoundError: If model configuration file is not found.
|
71 |
-
|
72 |
-
Returns:
|
73 |
-
CRS model.
|
74 |
-
"""
|
75 |
-
logging.debug(f"Loading CRS model {model_name}.")
|
76 |
-
if not os.path.exists(model_config_file):
|
77 |
-
raise FileNotFoundError(
|
78 |
-
f"Model configuration file {model_config_file} not found."
|
79 |
-
)
|
80 |
-
|
81 |
-
model_args = yaml.safe_load(open(model_config_file, "r"))
|
82 |
-
|
83 |
-
if "chatgpt" in model_name:
|
84 |
-
openai.api_key = st.secrets["openai_api_key"]
|
85 |
-
|
86 |
-
# Extract crs model from name
|
87 |
-
name = model_name.split("_")[0]
|
88 |
-
|
89 |
-
return CRSModel(name, **model_args)
|
90 |
|
91 |
|
92 |
def download_and_extract_models() -> None:
|
|
|
59 |
|
60 |
return CRSModel(name, **model_args), ttl=timedelta(days=3)
|
61 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
62 |
|
63 |
|
64 |
def download_and_extract_models() -> None:
|