Spaces:
Running
Running
João Pedro
commited on
Commit
·
ff1aca1
1
Parent(s):
d2efd6d
make feedback into a table for better vis + add btn
Browse files
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 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
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()
|