Merge pull request #2 from naseemap47/app

submit buttom only after updated image
naseemap47 · Jul 26, 2023 · 67ee33b · 67ee33b
2 parents ce819a4 + 0b7f6ff
commit 67ee33b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/app.py b/app.py
@@ -46,8 +46,8 @@ def single_image_inference(model, img):
     img_input = cv2.imdecode(file_bytes, 1)
     FRAME_WINDOW.image(img_input, channels='BGR')
 
-if st.button('Submit', use_container_width=True):
-    st.subheader('Result')
-    inp = img2tensor(img_input)
-    out_img = single_image_inference(NAFNet, inp)
-    st.image(out_img, channels='BGR')
+    if st.button('Submit', use_container_width=True):
+        st.subheader('Result')
+        inp = img2tensor(img_input)
+        out_img = single_image_inference(NAFNet, inp)
+        st.image(out_img, channels='BGR')