Fix gradio issue

app.py

@@ -507,15 +507,9 @@ with demo:
             with gr.Column():
                 gr.Markdown('## Input')
                 gr.Markdown('Upload the file with 3D-coordinates of the input fragments in .pdb, .mol2 or .sdf format:')
-                with gr.Column():
-                    input_fragments_file = gr.File(
-                        file_count='single',
-                        label='Input Fragments',
-                        file_types=['.sdf', '.pdb', '.mol', '.mol2']
-                    )
-                # gr.Markdown('(Optionally) upload the file of the target protein in .pdb format:')
-                with gr.Column():
-                    input_protein_file = gr.File(file_count='single', label='Target Protein', file_types=['.pdb'])
+                input_fragments_file = gr.File(file_count='single', label='Input Fragments')
+                gr.Markdown('(Optionally) upload the file of the target protein in .pdb format:')
+                input_protein_file = gr.File(file_count='single', label='Target Protein')
 
                 n_steps = gr.Slider(minimum=50, maximum=500, label="Number of Denoising Steps", step=10)
                 n_atoms = gr.Slider(
@@ -531,7 +525,7 @@ with demo:
                         ['examples/3hz1_fragments.sdf', 'examples/3hz1_protein.pdb'],
                         ['examples/5ou2_fragments.sdf', 'examples/5ou2_protein.pdb'],
                     ],
-                    headers=['Fragments', 'Target Protein'],
+                    # headers=['Fragments', 'Target Protein'],
                     type='values',
                 )