Spaces:

moldenhof
/

atomlenz

Runtime error

moldenhof commited on Mar 5, 2024

Commit

c6e0fcd

1 Parent(s): ffed2e4

implementing app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ if image_file is not None:
    dataset.prepare_data()
    trainer = pl.Trainer(logger=False)
    atom_preds = trainer.predict(model_atom, dataset.test_dataloader())
-##   st.write(atom_preds)
    plt.imshow(image, cmap="gray")
    for bbox, label in zip(atom_preds[0]['boxes'], atom_preds[0]['preds']):
        plot_bbox(bbox, label)

    dataset.prepare_data()
    trainer = pl.Trainer(logger=False)
    atom_preds = trainer.predict(model_atom, dataset.test_dataloader())
+   st.write(atom_preds)
    plt.imshow(image, cmap="gray")
    for bbox, label in zip(atom_preds[0]['boxes'], atom_preds[0]['preds']):
        plot_bbox(bbox, label)