jannisborn commited on
Commit
3d9aa37
·
unverified ·
1 Parent(s): 9eecb91
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -48,7 +48,7 @@ def run_inference(
48
  omic_path = None
49
 
50
  # ToDo: Add progress bar for multiple smiles
51
- results = {}
52
  for smi in tqdm(smiles, total=len(smiles)):
53
  result = submission(
54
  drug={"smiles": smi},
@@ -71,6 +71,7 @@ def run_inference(
71
  results[f"epistemic_confidence_{smi}"] = (
72
  result["aleatoric_confidence"].squeeze().round(3)
73
  )
 
74
  predicted_df = pd.DataFrame(results)
75
 
76
  # Prepare DF to visualize
 
48
  omic_path = None
49
 
50
  # ToDo: Add progress bar for multiple smiles
51
+ results = []
52
  for smi in tqdm(smiles, total=len(smiles)):
53
  result = submission(
54
  drug={"smiles": smi},
 
71
  results[f"epistemic_confidence_{smi}"] = (
72
  result["aleatoric_confidence"].squeeze().round(3)
73
  )
74
+ results.append(result)
75
  predicted_df = pd.DataFrame(results)
76
 
77
  # Prepare DF to visualize