Spaces:
Running
Running
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -289,7 +289,7 @@ class Inference(object):
|
|
289 |
generatedsmiles = pd.DataFrame({"SMILES": gen_smi})
|
290 |
generatedsmiles.to_csv("experiments/inference/{}/inference_drugs.csv".format(self.submodel), index=False)
|
291 |
|
292 |
-
|
293 |
|
294 |
if __name__=="__main__":
|
295 |
parser = argparse.ArgumentParser()
|
|
|
289 |
generatedsmiles = pd.DataFrame({"SMILES": gen_smi})
|
290 |
generatedsmiles.to_csv("experiments/inference/{}/inference_drugs.csv".format(self.submodel), index=False)
|
291 |
|
292 |
+
return model_res
|
293 |
|
294 |
if __name__=="__main__":
|
295 |
parser = argparse.ArgumentParser()
|