Update app.py

app.py

@@ -9,8 +9,22 @@ install_flash_attn()
 HF_TOKEN = os.getenv("HF_TOKEN")
 authenticate_hf(HF_TOKEN)
 
+# Define a customized theme based on the Soft theme
+theme = gr.themes.Soft(
+    primary_hue="blue",
+    secondary_hue="green",
+    neutral_hue="gray",
+    spacing_size="lg",
+    radius_size="sm",
+    text_size="lg"
+).set(
+    button_primary_background_fill="*primary_500",
+    button_primary_background_fill_hover="*primary_700",
+    button_primary_border_radius="*radius_md"
+)
+
 # Create the Gradio Blocks interface
-with gr.Blocks(css="styles.css") as demo:
+with gr.Blocks(theme=theme) as demo:
     with gr.Row():
         with gr.Column():
             textbox = gr.Textbox(label="Model Name", placeholder="Enter the model name here OR select example below...", lines=1)
@@ -45,8 +59,8 @@ with gr.Blocks(css="styles.css") as demo:
             )
             submit_button = gr.Button("Submit")
         with gr.Column():
-            output = gr.Textbox(label="Model Architecture", lines=20, placeholder="Model architecture will appear here...", show_copy_button=True, elem_classes=["model-architecture"])
-            error_output = gr.Textbox(label="Error", lines=10, placeholder="Exceptions will appear here...", show_copy_button=True, elem_classes=["error-box"])
+            output = gr.Textbox(label="Model Architecture", lines=20, placeholder="Model architecture will appear here...", show_copy_button=True)
+            error_output = gr.Textbox(label="Error", lines=10, placeholder="Exceptions will appear here...", show_copy_button=True)
 
     def handle_click(model_name):
         model_summary, error_message = get_model_summary(model_name)