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