update

app.py

@@ -51,7 +51,6 @@ def run_inference(
 
     result = pd.DataFrame({})
     for smi in tqdm(smiles, total=len(smiles)):
-        print("PARSING", smi)
         output = submission(
             drug={"smiles": smi},
             workspace_id="emulated_workspace_id",
@@ -89,7 +88,7 @@ def run_inference(
         )
         df["site"] = df["site"].apply(lambda x: site_mapper.get(x, x))
         df["cell_line"] = df["cell_line"].apply(lambda x: x.split("_")[0])
-        if confidence:
+        if not confidence:
             df.drop(
                 ["aleatoric_confidence", "epistemic_confidence"], axis=1, inplace=True
             )