add explanation to app

.streamlit/config.toml

@@ -1 +1,3 @@
+[server]
+
 maxUploadSize = 1

app.py

@@ -8,18 +8,12 @@ import jax.numpy as jnp
 import os
 import jax
 
-st.header('Under construction')
+# st.header('Under construction')
 
-
-st.title("CLIP Reply Demo")
-st.sidebar.markdown(
-    """   
-
-    Validation set:  351 images/273 deduped (There are still duplicates)
-
-    Example Queries : 
-"""
-)
+st.sidebar.write("")
+st.title("CLIP React Demo")
+st.write("[Model Card](https://huggingface.co/flax-community/clip-reply)")
+st.write("    ")
 @st.cache(allow_output_mutation=True)
 def load_model():
     model = FlaxHybridCLIP.from_pretrained("ceyda/clip-reply")
@@ -41,7 +35,7 @@ image_index = load_image_index()
 model, processor = load_model()
 
 col_count=4
-top_k=10
+top_k=st.sidebar.slider("Show top-K", min_value=1, max_value=50, value=20)
 
 show_val=st.sidebar.button("show all validation set images")
 if show_val:
@@ -76,15 +70,35 @@ def query_with_images(query_images,query_text):
     st.write(results)
     return zip(*results)
 
-q_cols=st.beta_columns(2)    
-query_text = q_cols[0].text_input("Input text", value="I love you")
-query_images = q_cols[1].file_uploader("(optional) upload query image",type=['jpg','jpeg'], accept_multiple_files=True)
+q_cols=st.beta_columns([5,2,5])    
+
+q_cols[0].markdown(
+"""
+    Example Queries : 
+
+     - I'm so scared right now
+     - I got the job 🎉
+     - OMG that is disgusting 
+     - I'm awesome
+    
+"""
+)
+q_cols[2].markdown(
+    """   
+    Searches among the validation set images if not specified
+    (There may be non-exact duplicates)
+
+"""
+)
+
+query_text = q_cols[0].text_input("Input text you want to get reaction for", value="I love you ❤️")
+query_images = q_cols[2].file_uploader("(optional) Upload images to rank them",type=['jpg','jpeg'], accept_multiple_files=True)
 
 if query_images:
-    st.write("Ranking uploaded images with respect to input text")
+    st.write("Ranking your uploaded images with respect to input text:")
     ids, dists = query_with_images(query_images,query_text)
 else:
-    st.write("Finding within validation set")
+    st.write("Found these images within validation set:")
     proc = processor(text=[query_text], images=None, return_tensors="jax", padding=True)
     vec = np.asarray(model.get_text_features(**proc))
     ids, dists = image_index.knnQuery(vec, k=top_k)
@@ -96,8 +110,8 @@ for i,(id_, dist) in enumerate(zip(ids, dists)):
         if isinstance(id_, np.int32):
             st.image("./imgs/"+file_names[id_])
             # st.write(file_names[id_])
-            st.write(1.0 - dist)
+            st.write(1.0 - dist, help="score")
         else:
             st.image(id_)
-            st.write(dist)
+            st.write(dist, help="score")