update app

app.py

@@ -1,3 +1,6 @@
+# Developed by Mohammad Khalooei
+# More information and contact: https://github.com/khalooei/LSA
+
 import torch
 import torch.nn as nn
 import torchvision
@@ -12,7 +15,6 @@ import time
 from datetime import datetime
 import gradio as gr
 
-# LeNet for MNIST
 class LeNet(nn.Module):
     def __init__(self):
         super(LeNet, self).__init__()
@@ -138,7 +140,7 @@ def get_dataset_and_transform(dataset_name):
             transforms.Normalize((0.1307,), (0.3081,))
         ])
         dataset = torchvision.datasets.MNIST(root='./data', train=False, download=True, transform=transform)
-    else:  # CIFAR-10
+    else:
         transform = transforms.Compose([
             transforms.Resize((224, 224)),
             transforms.ToTensor(),
@@ -173,7 +175,7 @@ def initialize_model(model_name, device):
 
 def layer_sustainability_analysis(dataset_name, model_name, selected_attacks, num_batches, output_dir_base='outputs'):
     start_time = time.time()
-    logs = []
+    logs = ["BSM:: experiment is being started ..."]
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
     dataset, _ = get_dataset_and_transform(dataset_name)
@@ -295,7 +297,7 @@ def layer_sustainability_analysis(dataset_name, model_name, selected_attacks, nu
         mvl_plot_paths.append(None)
 
     return [
-        None,  # no error
+        None,
         cm_plot_path,
         *mvl_plot_paths[:5],
         integrated_mvl_plot_path,
@@ -340,9 +342,9 @@ def create_interface():
         gr.Markdown("# Layer-wise Sustainability Analysis")
         gr.Markdown(paper_info_html)
 
-        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")
+        initial_input="MNIST"
+        dataset_input = gr.Dropdown(datasets, label="Select Dataset", value=initial_input)
+        model_input = gr.Dropdown(get_models_for_dataset(initial_input), value=get_models_for_dataset(initial_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)
@@ -363,7 +365,7 @@ def create_interface():
             with gr.Tab("Model Statistics"):
                 stats_output = gr.Markdown("## Model Statistics")
             with gr.Tab("Logs"):
-                log_output = gr.Textbox(label="Processing Logs")
+                log_output = gr.Textbox(label="Processing Logs", lines=15, interactive=False)
 
         dataset_input.change(
             fn=update_models,