Update app.py

app.py

@@ -12,15 +12,18 @@ def add_model_to_list(new_model):
         model_list.append(new_model)  # Add the new model to the list.
     return model_list
 
-
 # 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
-    
+
+    # Ensure num_labels is an integer
+    num_labels = int(num_labels)
+
     # 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)
+
     # Create a configuration for the selected model using AutoConfig.
     config = AutoConfig.from_pretrained(model_name, num_labels=num_labels, use_cache=use_cache)
     return str(config)  # Return the configuration as a string.