Update gradio_app.py

gradio_app.py

@@ -106,8 +106,12 @@ def function(model_name: str, mol_num: int, seed: int) -> tuple[PIL.Image, pd.Da
     scores = trainer.inference() # create scores_df out of this
 
     score_df = pd.DataFrame(scores, index=[0])
-
-    output_file_path = f'experiments/inference/{model_name}/inference_drugs.txt'
+    old_model_names = {
+        "DrugGEN": "CrossLoss",
+        "DrugGEN-NoTarget": "NoTarget",
+    }
+    output_file_path = f'experiments/inference/{old_model_names[model_name]}/inference_drugs.txt'
+    
     import os
     new_path = f'{model_name}_denovo_mols.smi'
     os.rename(output_file_path, new_path)