Update app.py

app.py

@@ -91,8 +91,19 @@ if uploaded_file is not None:
         st.write(f"Recognized Character Sequence: {output}")
 
 st.write("## examples")
-cols = st.columns(4)
+# cols = st.columns(4)
+# for idx,(model_variant,captcha_path) in enumerate(example_captchas.items()):
+#     col = cols[idx % 4]
+#     col.image(captcha_path,caption=f'{captcha_path.split("/")[-1]}', width=100)
+#     col.write(f"Model Version: {model_variant}")
+
 for idx,(model_variant,captcha_path) in enumerate(example_captchas.items()):
-    col = cols[idx % 4]
-    col.image(captcha_path,caption=f'{captcha_path.split("/")[-1]}', width=100)
-    col.write(f"Model Version: {model_variant}")
+    with cols[idx % 4]:
+        st.image(captcha_path, caption=f'{captcha_path.split("/")[-1]}', width=130)
+        if st.button(f"Recognize with {model_variant}", key=f"button_{idx}"):
+            input = transform_image(captcha_path)
+            model = models[model_variant]
+            with torch.no_grad():
+                model_prediction = model(input)
+                output = get_label(model_prediction, model_variant)
+            st.write(f"Model {model_variant} recognized: {output}")