Spaces:

HUBioDataLab
/

DrugGEN

Running

mgyigit commited on Mar 15

Commit

d7fd6f0

verified ·

1 Parent(s): c724df9

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -289,7 +289,7 @@ class Inference(object):
         generatedsmiles = pd.DataFrame({"SMILES": gen_smi})
         generatedsmiles.to_csv("experiments/inference/{}/inference_drugs.csv".format(self.submodel), index=False)
-    return model_res
 if __name__=="__main__":
     parser = argparse.ArgumentParser()

         generatedsmiles = pd.DataFrame({"SMILES": gen_smi})
         generatedsmiles.to_csv("experiments/inference/{}/inference_drugs.csv".format(self.submodel), index=False)
+        return model_res
 if __name__=="__main__":
     parser = argparse.ArgumentParser()