Update app.py

app.py

@@ -88,6 +88,7 @@ class model:
 base_class = model()
 
 with gr.Blocks(theme='gradio/soft') as demo:
+    gr.Markdown("## Inputs")
     gr.Markdown("Select the prediction model. **Note the first run may take ~1-2 minutes, but will take 2-3 seconds afterwards.**")
     gr.Markdown(
         "```CELL-E_2_HPA_480``` is a good general purpose model for various cell types using ICC-IF."
@@ -126,6 +127,9 @@ with gr.Blocks(theme='gradio/soft') as demo:
 
         protein_image = gr.Image(type="pil", label="Protein Image (Optional)")
 
+    with gr.Row():
+        gr.Markdown("## Outputs")
+
     with gr.Row():
         gr.Markdown("Image predictions are show below.")