Update app.py

app.py

@@ -76,13 +76,13 @@ class model:
  device=device,
  )
  
- return T.ToPILImage()(protein_image[0,0]), T.ToPILImage()(nucleus_image[0,0]), formatted_predicted_sequence
+ return T.ToPILImage()(protein_image[0,0]), T.ToPILImage()(nucleus_image[0,0]), formatted_predicted_sequence.replace("<pad>", "")
 
 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 ~5-6 minutes, but will take 3-4 seconds afterwards.**")
+ gr.Markdown("Select the prediction model. **Note the first run may take ~2-3 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."
  )