Update app.py

app.py

@@ -15,6 +15,9 @@ def add_model_to_list(new_model):
 
 # Function to create a configuration for the selected model.
 def create_config(model_name, num_labels, use_cache):
+    if isinstance(model_name, list):
+        model_name = model_name[0]  # Take the first model name from the list if it's a list
+    
     # If the selected model is not in the list, add it (this is a safety check).
     if model_name not in model_list:
         model_list.append(model_name)