Update app.py

app.py

@@ -300,19 +300,13 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
             with gr.Row():
                 with gr.Column(scale=1):
                     # Reuse model selection for custom input
-                    model_name_custom = gr.Radio(
-                        choices=("DrugGEN-AKT1", "DrugGEN-CDK2", "DrugGEN-NoTarget"),
-                        value="DrugGEN-AKT1",
-                        label="Select Target Model",
-                        info="Choose which protein target or general model to use for evaluation"
-                    )
                     custom_smiles = gr.Textbox(
                         label="Input SMILES (one per line, maximum 100 molecules)",
                         placeholder="C(C(=O)O)N\nCCO\n...",
                         lines=10
                     )
                     custom_submit = gr.Button(
-                        value="Evaluate Custom SMILES",
+                        value="Generate Molecules using Custom SMILES",
                         variant="primary",
                         size="lg"
                     )
@@ -350,12 +344,12 @@ For more details, see our [paper on arXiv](https://arxiv.org/abs/2302.07868).
     
     custom_submit.click(
         run_inference, 
-        inputs=[gr.State("Custom Input SMILES"), model_name_custom, gr.State(0), gr.State(""), custom_smiles],
+        inputs=[gr.State("Custom Input SMILES"), model_name, gr.State(0), gr.State(""), custom_smiles],
         outputs=[
-            image_output_custom, 
-            file_download_custom,
-            basic_metrics_df_custom,
-            advanced_metrics_df_custom
+            image_output, 
+            file_download,
+            basic_metrics_df,
+            advanced_metrics_df
         ],
         api_name="inference_custom"
     )