Spaces:

anasanchezf
/

cloome

Sleeping

Sanchez fernandez commited on May 12, 2023

Commit

3c88f48

1 Parent(s): ed68314

Include similarity score

Files changed (1) hide show

app.py CHANGED Viewed

@@ -480,7 +480,7 @@ def images_from_molecule(top_n, model_path):
                     im = reshape_image(image)
                     index = n+1
-                    score = top_probs[n] * 100
                     col.image(im, caption=f"Top {index}. Score: {round(score)}", width=200)

                     im = reshape_image(image)
                     index = n+1
+                    score = float(top_probs[n]) * 100
                     col.image(im, caption=f"Top {index}. Score: {round(score)}", width=200)