Just highlight

src/app.py

@@ -16,24 +16,27 @@ def change(inp, components=[]):
     """
     code, explanation, docs = get_text(inp)
     if inp == "Basic":
-        return (templates["initial"], highlight(code), "## Accelerate Code (Base Integration)", explanation, docs)
+        return (highlight(code), "## Accelerate Code (Base Integration)", explanation, docs)
+        # return (templates["initial"], highlight(code), "## Accelerate Code (Base Integration)", explanation, docs)
     elif inp == "Calculating Metrics":
-        return (templates["initial_with_metrics"], highlight(code), f"## Accelerate Code ({inp})", explanation, docs)
+        return (highlight(code), f"## Accelerate Code ({inp})", explanation, docs)
+        # return (templates["initial_with_metrics"], highlight(code), f"## Accelerate Code ({inp})", explanation, docs)
     else:
-        return (templates["accelerate"], highlight(code), f"## Accelerate Code ({inp})", explanation, docs)
+        return (highlight(code), f"## Accelerate Code ({inp})", explanation, docs)
+        # return (templates["accelerate"], highlight(code), f"## Accelerate Code ({inp})", explanation, docs)
 
 initial_md = gr.Markdown("## Initial Code")
 initial_code = gr.Markdown(templates["initial"])
 
-with gr.Blocks() as demo:
+with gr.Blocks(css=".gradio-container {padding: 5px}") as demo:
     inp = gr.Radio(
         ["Basic", "Calculating Metrics", "Checkpointing", "Experiment Tracking", "Gradient Accumulation"], 
         label="Select a feature you would like to integrate"
     )
-    with gr.Row():
-        with gr.Column():
-            initial_md.render()
-            initial_code.render()
+    # with gr.Row():
+    #     with gr.Column():
+    #         initial_md.render()
+    #         initial_code.render()
     with gr.Row():
         with gr.Column():
             feature = gr.Markdown("## Accelerate Code")
@@ -48,7 +51,8 @@ with gr.Blocks() as demo:
             docs = gr.Markdown()
     inp.change(
         fn=change, 
-        inputs=inp, 
-        outputs=[initial_code, out, feature, explanation, docs]
+        inputs=inp,
+        outputs=[out, feature, explanation, docs] 
+        # outputs=[initial_code, out, feature, explanation, docs]
     )
 demo.launch()