Update gradio_app.py

gradio_app.py

@@ -25,7 +25,7 @@ class DrugGENConfig:
     heads=8
     mlp_ratio=3
     dropout=0.
-    log_sample_step=1000
+    log_sample_step=100
     set_seed=True
     seed=10
 
@@ -43,15 +43,15 @@ model_configs = {
 
 
 
-def function(model_name: str, mol_num: int, seed: int) -> tuple[PIL.Image, pd.DataFrame, str]:
+def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Image, pd.DataFrame, str]:
     '''
     Returns:
     image, score_df, file path
     '''
 
     config = model_configs[model_name]
-    config.inference_sample_num = mol_num
-    config.seed = seed
+    config.inference_sample_num = num_molecules
+    config.seed = seed_num
 
     inferer = Inference(config)
     scores = inferer.inference() # create scores_df out of this