Update app.py

app.py

@@ -79,12 +79,13 @@ class model:
 base_class = model()
 
 with gr.Blocks(theme='gradio/soft') as demo:
-    gr.Markdown("Select the prediction model.")
+    gr.Markdown("## Inputs")
+    gr.Markdown("Select the prediction model. **Note the first run may take ~5-6 minutes, but will take 3-4 seconds afterwards.**")
     gr.Markdown(
-        "- CELL-E_2_HPA_2560 is a good general purpose model for various cell types using ICC-IF."
+        "- ```CELL-E_2_HPA_2560``` is a good general purpose model for various cell types using ICC-IF."
     )
     gr.Markdown(
-        "- CELL-E_2_OpenCell_2560 is trained on OpenCell and is good more live-cell predictions on HEK cells."
+        "- ```CELL-E_2_OpenCell_2560``` is trained on OpenCell and is good more live-cell predictions on HEK cells."
     )
     with gr.Row():
         model_name = gr.Dropdown(
@@ -117,8 +118,10 @@ with gr.Blocks(theme='gradio/soft') as demo:
             image_mode="L",
             type="pil"
         )
-
-
+        
+    with gr.Row():
+        gr.Markdown("## Outputs")
+        
     with gr.Row().style(equal_height=True):
         nucleus_crop = gr.Image(
             label="Nucleus Image (Crop)",