João Pedro commited on
Commit
ff1aca1
·
1 Parent(s): d2efd6d

make feedback into a table for better vis + add btn

Browse files
Files changed (1) hide show
  1. app.py +18 -8
app.py CHANGED
@@ -41,6 +41,12 @@ uploaded_file = st.file_uploader(
41
  "Upload Document", type=["pdf", "jpg", "png"], accept_multiple_files=False
42
  )
43
 
 
 
 
 
 
 
44
  if uploaded_file:
45
  run = wandb.init(project='hydra-classifier', name='feedback-loop')
46
 
@@ -77,13 +83,17 @@ if uploaded_file:
77
  )
78
  print(f'Correct label for image {i}: {correct_label}')
79
 
80
- run.log({
81
- 'image': wandb.Image(image),
82
- 'filetype': uploaded_file.type,
83
- 'predicted_label': id2label[prediction],
84
- 'predicted_label_id': prediction,
85
- 'correct_label': correct_label,
86
- 'correct_label_id': label2id[correct_label]
87
- })
 
 
 
88
 
 
89
  run.finish()
 
41
  "Upload Document", type=["pdf", "jpg", "png"], accept_multiple_files=False
42
  )
43
 
44
+
45
+ feedback_table = wandb.Table(columns=[
46
+ 'image', 'filetype', 'predicted_label', 'predicted_label_id',
47
+ 'correct_label', 'correct_label_id'
48
+ ])
49
+
50
  if uploaded_file:
51
  run = wandb.init(project='hydra-classifier', name='feedback-loop')
52
 
 
83
  )
84
  print(f'Correct label for image {i}: {correct_label}')
85
 
86
+ # Add a button to confirm feedback and log it
87
+ if st.button(f"Submit Feedback for Image {i}", key=f'submit-{i}'):
88
+ feedback_table.add({
89
+ 'image': wandb.Image(image),
90
+ 'filetype': uploaded_file.type,
91
+ 'predicted_label': id2label[prediction],
92
+ 'predicted_label_id': prediction,
93
+ 'correct_label': correct_label,
94
+ 'correct_label_id': label2id[correct_label]
95
+ })
96
+ st.success(f"Feedback for Image {i} submitted!")
97
 
98
+ run.log({'feedback_table': feedback_table})
99
  run.finish()