implementing app

app.py

@@ -65,7 +65,7 @@ if image_file is not None:
    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'].numpy(), atom_preds[0]['preds'].numpy()):
+   for bbox, label in zip(atom_preds[0]['boxes'][0], atom_preds[0]['preds'][0]):
        st.write(bbox)
        st.write(label)
        plot_bbox(bbox, label)