MattGPT commited on
Commit
08d6f20
1 Parent(s): a16c241

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -15,6 +15,9 @@ def add_model_to_list(new_model):
15
 
16
  # Function to create a configuration for the selected model.
17
  def create_config(model_name, num_labels, use_cache):
 
 
 
18
  # If the selected model is not in the list, add it (this is a safety check).
19
  if model_name not in model_list:
20
  model_list.append(model_name)
 
15
 
16
  # Function to create a configuration for the selected model.
17
  def create_config(model_name, num_labels, use_cache):
18
+ if isinstance(model_name, list):
19
+ model_name = model_name[0] # Take the first model name from the list if it's a list
20
+
21
  # If the selected model is not in the list, add it (this is a safety check).
22
  if model_name not in model_list:
23
  model_list.append(model_name)