Update app.py

app.py

@@ -434,7 +434,7 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
             """)
             
             # Create container for generation mode inputs
-            with gr.Box(visible=True, elem_id="generate-box", elem_classes="input-box") as generate_group:
+            with gr.Group(visible=True, elem_id="generate-box", elem_classes="input-box") as generate_group:
                 num_molecules = gr.Slider(
                     minimum=10,
                     maximum=250,
@@ -452,7 +452,7 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
                 )
             
             # Create container for SMILES input mode
-            with gr.Box(visible=False, elem_id="smiles-box", elem_classes="input-box") as smiles_group:
+            with gr.Group(visible=False, elem_id="smiles-box", elem_classes="input-box") as smiles_group:
                 smiles_input = gr.Textbox(
                     label="Input SMILES",
                     info="Enter up to 100 SMILES strings, one per line",