Nol00 commited on
Commit
d64e52a
·
verified ·
1 Parent(s): 1e5f915

Update crs_arena/utils.py

Browse files
Files changed (1) hide show
  1. 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: