Remove extra py3Dmol options

main.py

@@ -193,9 +193,10 @@ def predict(pdb_code, pdb_file):
     view.setBackgroundColor('white')
     view.addModel(pdb, "pdb")
     view.setStyle({'stick': {'colorscheme': {'prop': 'resi', 'C': 'turquoise'}}})
-    view.addLight([0, 0, 10], [1, 1, 1], 1)  # Add directional light from the z-axis
-    view.setSpecular(0.5)  # Adjust the specular lighting effect
-    view.setAmbient(0.5)  # Adjust the ambient lighting effect
+    # Commenting since the visualizer is not rendered
+    # view.addLight([0, 0, 10], [1, 1, 1], 1)  # Add directional light from the z-axis
+    # view.setSpecular(0.5)  # Adjust the specular lighting effect
+    # view.setAmbient(0.5)  # Adjust the ambient lighting effect
     
     for i in range(10):
         adaptability_value = adaptability[topN_ind[i]]
@@ -233,6 +234,9 @@ def run():
         #text_output = gr.Textbox()
         #text_button = gr.Button("Flip")
         inp = gr.Textbox(placeholder="PDB Code or upload file below", label="Input structure")
+        topN = gr.Slider(value=10,
+            minimum=1, maximum=100, label="Number of atoms to visualize", step=1
+        )
         pdb_file = gr.File(label="PDB File Upload")
         #with gr.Row():
         #    helix = gr.ColorPicker(label="helix")