implementing app
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ if image_file is not None:
|
|
64 |
atom_preds = trainer.predict(model_atom, dataset.test_dataloader())
|
65 |
st.write(atom_preds)
|
66 |
plt.imshow(image, cmap="gray")
|
67 |
-
for bbox, label in zip(atom_preds[0]['boxes'], atom_preds[0]['preds']):
|
68 |
plot_bbox(bbox, label)
|
69 |
|
70 |
col2.image(plt.show(), use_column_width=True)
|
|
|
64 |
atom_preds = trainer.predict(model_atom, dataset.test_dataloader())
|
65 |
st.write(atom_preds)
|
66 |
plt.imshow(image, cmap="gray")
|
67 |
+
for bbox, label in zip(atom_preds[0]['boxes'][0], atom_preds[0]['preds']):
|
68 |
plot_bbox(bbox, label)
|
69 |
|
70 |
col2.image(plt.show(), use_column_width=True)
|