Update app.py

app.py

@@ -230,22 +230,22 @@ metal3d = gr.Blocks()
 
 with metal3d:
     gr.Markdown("# Metal3D")
-    gr.Markdown(
-        """
-        Details about implementation and code available here:
-        >Duerr, Levy and Roethlisberger, Predicting zinc ion location using deep learning, BioRxiv, 2022 "
-    """
-    )
+
     with gr.Group():
         inp = gr.Textbox(
-            placeholder="PDB Code or Uniprot identifier", label="Input molecule"
+            placeholder="PDB Code or Uniprot identifier or upload file below", label="Input molecule"
         )
-        gr.Markdown("or upload a file")
         file = gr.File(file_count="single", type="file")
-        mode = gr.Radio(
-            ["All metalbinding residues (ASP, CYS, GLU, HIS)", "All residues"],
-            label="Residues to use for prediction",
-        )
+        with gr.Row():
+            mode = gr.Radio(
+                ["All metalbinding residues (ASP, CYS, GLU, HIS)", "All residues"],
+                label="Residues to use for prediction",
+            )
+            custom_resids = gr.Textbox(placeholder="Comma separated list of residues", label="Custom residues")
+        gr.Markdown(
+        """ <small>Inference using CPU-only, can be quite slow for more than 20 residues. Use Colab notebook for GPU acceleration</small>
+    """
+    )
         btn = gr.Button("Run")
 
     gr.Markdown("# Output")