update

app.py

@@ -67,10 +67,10 @@ def run_inference(
         result[f"IC50_{smi}"].shape
         if confidence:
             result[f"aleatoric_confidence_{smi}"] = (
-                result["aleatoric_confidence"].squeeze().round(3)
+                output["aleatoric_confidence"].squeeze().round(3)
             )
             result[f"epistemic_confidence_{smi}"] = (
-                result["aleatoric_confidence"].squeeze().round(3)
+                output["aleatoric_confidence"].squeeze().round(3)
             )
         results.append(result)
     predicted_df = pd.DataFrame(results)