Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def create_config(model_name, num_labels, use_cache):
|
|
25 |
model_list.append(model_name)
|
26 |
|
27 |
# Create a configuration for the selected model using AutoConfig.
|
28 |
-
config = AutoConfig.from_pretrained(model_name, num_labels=num_labels, use_cache=
|
29 |
return str(config) # Return the configuration as a string.
|
30 |
|
31 |
# Start building the Gradio interface
|
|
|
25 |
model_list.append(model_name)
|
26 |
|
27 |
# Create a configuration for the selected model using AutoConfig.
|
28 |
+
config = AutoConfig.from_pretrained(model_name, num_labels=num_labels, use_cache='true')
|
29 |
return str(config) # Return the configuration as a string.
|
30 |
|
31 |
# Start building the Gradio interface
|