Update gradio_app.py

gradio_app.py

@@ -39,7 +39,7 @@ class NoTargetConfig(DrugGENConfig):
 
 model_configs = {
     "DrugGEN": DrugGENConfig(),
-    "DrugGEN-NoTarget": NoTargetConfig(),
+    "NoTarget": NoTargetConfig(),
 }
 
 
@@ -49,7 +49,9 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
     Returns:
     image, score_df, file path
     '''
-
+    if model_name == "DrugGEN-NoTarget":
+        model_name = "NoTarget"
+    
     config = model_configs[model_name]
     config.sample_num = num_molecules