moldenhof commited on
Commit
134737a
·
1 Parent(s): c6e0fcd

implementing app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)