Update gradio_app.py

gradio_app.py

@@ -8,6 +8,7 @@ from rdkit import Chem
 from rdkit.Chem import Draw
 from rdkit.Chem.Draw import IPythonConsole
 import shutil
+import os
 
 class DrugGENConfig:
     submodel='DrugGEN'
@@ -52,15 +53,15 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
     config = model_configs[model_name]
     config.sample_num = num_molecules
 
-    print(seed_num)
     if seed_num is not None:
         try:
             config.seed = int(seed_num)
-        except:
-            raise Exception("The seed must be an integer value!")
+        except ValueError: 
+            raise gr.Warning("The seed must be an integer value!")
     else:
         config.seed = random.randint(0, 10000)
 
+
     inferer = Inference(config)
     scores = inferer.inference() # create scores_df out of this
 
@@ -68,7 +69,6 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
 
     output_file_path = f'experiments/inference/{model_name}/inference_drugs.txt'
 
-    import os
     new_path = f'{model_name}_denovo_mols.smi'
     os.rename(output_file_path, new_path)
 
@@ -135,7 +135,8 @@ with gr.Blocks() as demo:
             )
 
             seed_num = gr.Textbox(
-                label="RNG seed value (can be used for reproducibility):"
+                label="RNG seed value (can be used for reproducibility):",
+                value=None
             )
 
             submit_button = gr.Button(