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)",