Update app.py

app.py

@@ -13,13 +13,16 @@ os.environ['CUDA_VISIBLE_DEVICES'] = ''
 
 
 st.markdown("## Finder!")
-st.markdown("### Get a segmentation mask by a text describtion or a reference image!")
+st.markdown("### Get a segmentation mask by a text describtion or a reference image.")
 st.markdown("DISCLAIMER: this is working much longer because of cpu :(")
 st.markdown("<img width=200px src='https://rozetked.me/images/uploads/dwoilp3BVjlE.jpg'>", unsafe_allow_html=True)
 # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
 
 
-uploaded_file = st.file_uploader("IMAGE HERE", type=['png', 'jpg', 'jpeg'])
+uploaded_file = st.file_uploader(
+    "Provide an image where you want to find something", 
+    type=['png', 'jpg', 'jpeg']
+    )
 
 
 
@@ -120,26 +123,26 @@ def get_top_mask(image_bytes, text=None, ref_image=None):
 
 if uploaded_file is not None:
     image_bytes = uploaded_file.getvalue()
-    option = st.selectbox('Find by?', ('Text', 'Image'))
+    option = st.selectbox('Find a segmentation mask by what mode of reference?', ('Text', 'Image'))
     to_show = Image.open(io.BytesIO(uploaded_file.getvalue()))
 
     text = ""
     ref_image_file = None
     
     if option == 'Text':
-        text = st.text_input('Searching for')
+        text = st.text_input('Textual describtion of an object of interest')
         _, mask  = get_top_mask(image_bytes, text=text)
     elif option == 'Image':
-        ref_image_file = st.file_uploader("REFERENCE IMAGE HERE", type=['png', 'jpg', 'jpeg'])
+        ref_image_file = st.file_uploader("Reference image of an object of interest", type=['png', 'jpg', 'jpeg'])
         if ref_image_file:
             ref_image_bytes = ref_image_file.getvalue()
             ref_image = Image.open(io.BytesIO(ref_image_bytes))
-            st.image(ref_image, caption='YOUR REFERENCE')
+            st.image(ref_image, caption='Your reference image')
             _, mask  = get_top_mask(image_bytes, ref_image=ref_image)
 
     if (option == 'Text' and text != "") or (option=='Image' and (ref_image_file is not None)):
         to_show = Image.fromarray(
             np.clip(np.array([1, 0, 0]) * mask[:, :, None] * 100 + np.array(to_show, dtype=int), a_min = 0, a_max = 255).astype(np.uint8)
             )
-    st.image(to_show, caption='Your uploaded image')
+    st.image(to_show, caption='Input image with highlighted mask')