Update app.py

add elem_classes properties for styling.

app.py

@@ -45,8 +45,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)
-            error_output = gr.Textbox(label="Error", lines=10, placeholder="Exceptions will appear here...", show_copy_button=True)
+            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"])
 
     def handle_click(model_name):
         model_summary, error_message = get_model_summary(model_name)