remove debug

app.py

@@ -34,6 +34,9 @@ with st.sidebar.beta_expander("Motivation",expanded=True):
     )
 top_k=st.sidebar.slider("Show top-K", min_value=1, max_value=50, value=20)
 col_count=4
+file_names=os.listdir("./imgs")
+file_names.sort()
+
 show_val=st.sidebar.button("show all validation set images")
 if show_val:
     cols=st.sidebar.beta_columns(col_count)
@@ -58,8 +61,7 @@ def load_image_index():
 
     return index
 
-file_names=os.listdir("./imgs")
-file_names.sort()
+
 
 image_index = load_image_index()
 model, processor = load_model()
@@ -85,17 +87,17 @@ def query_with_images(query_images,query_text):
     inputs["pixel_values"] = jnp.transpose(inputs["pixel_values"], axes=[0, 2, 3, 1])
     outputs = model(**inputs)
     logits_per_image = outputs.logits_per_image.reshape(-1)
-    st.write(logits_per_image)
+    # st.write(logits_per_image)
     probs = jax.nn.softmax(logits_per_image)
-    st.write(probs)
-    st.write(list(zip(images,probs)))
+    # st.write(probs)
+    # st.write(list(zip(images,probs)))
     results = sorted(list(zip(images,probs)),key=lambda x: x[1], reverse=True)
-    st.write(results)
+    # st.write(results)
     return zip(*results)
 
 q_cols=st.beta_columns([5,2,5])    
 
-examples = ["I'm so scared right now"," I got the job 🎉","OMG that is disgusting","You're awesome","I love you ❤️"]
+examples = ["OMG that is disgusting","I'm so scared right now"," I got the job 🎉","Congratulations to all the flax-community week teams","You're awesome","I love you ❤️"]
 example_input = q_cols[0].radio("Example Queries :",examples,index=4)
 q_cols[2].markdown(
     """