Update app.py

app.py

@@ -74,8 +74,6 @@ def function(model_name: str, num_molecules: int, seed_num: int):
     '''
     if model_name == "DrugGEN-NoTarget":
         model_name = "NoTarget"
-    else:
-        model_name = "DrugGEN"
     
     config = model_configs[model_name]
     config.sample_num = num_molecules
@@ -91,6 +89,9 @@ def function(model_name: str, num_molecules: int, seed_num: int):
         except ValueError:
             raise gr.Error("The seed must be an integer value!")
 
+    if model_name != "NoTarget":
+        model_name = "DrugGEN"
+
     inferer = Inference(config)
     start_time = time.time()
     scores = inferer.inference()  # This returns a DataFrame with specific columns