update app

app.py

@@ -340,8 +340,9 @@ def create_interface():
         gr.Markdown("# Layer-wise Sustainability Analysis")
         gr.Markdown(paper_info_html)
 
-        dataset_input = gr.Dropdown(datasets, label="Select Dataset", value='CIFAR-10')
-        model_input = gr.Dropdown(get_models_for_dataset('CIFAR-10'), value=get_models_for_dataset('CIFAR-10')[0], label="Select Model")
+        default_input="MNIST"
+        dataset_input = gr.Dropdown(datasets, label="Select Dataset", value=default_input)
+        model_input = gr.Dropdown(get_models_for_dataset(default_input), value=get_models_for_dataset(default_input)[0], label="Select Model")
         model_text = gr.Textbox(value="LeNet", visible=False, interactive=False, label="Model")
 
         attack_input = gr.CheckboxGroup(choices=attacks, label="Select Attacks", value=attacks)