fixing a few things

image2text.py

@@ -13,7 +13,8 @@ def app():
 
         ### 👋 Ciao!
 
-        Here you can find the captions that are most related to a given image.
+        Here you can find the captions or the labels that are most related to a given image. It is a zero-shot
+        image classification task!
 
         🤌 Italian mode on! 🤌
 
@@ -30,20 +31,20 @@ def app():
 
     with col2:
         captions_count = st.selectbox(
-            "Number of captions", options=range(1, MAX_CAP + 1)
+            "Number of labels", options=range(1, MAX_CAP + 1)
         )
         compute = st.button("Compute")
 
     with col1:
         captions = list()
         for idx in range(min(MAX_CAP, captions_count)):
-            captions.append(st.text_input(f"Insert Caption {idx+1}"))
+            captions.append(st.text_input(f"Insert label {idx+1}"))
 
     if compute:
         captions = [c for c in captions if c != ""]
 
         if not captions or not filename:
-            st.error("Please choose one image and at least one caption")
+            st.error("Please choose one image and at least one label")
         else:
             with st.spinner("Computing..."):
                 model = get_model()

introduction.md

@@ -150,14 +150,14 @@ then there is its (partial) counting ability and finally the ability of understa
 Look at the following - slightly cherry picked (but not even that much) - examples:
 
 ### Colors
-Here's a blu flower
+Here's a yellow flower
 <img src="https://huggingface.co/spaces/clip-italian/clip-italian-demo/raw/main/static/img/fiore_giallo.png" alt="drawing" width="600"/>
 
-And here's a yellow flower
+And here's a blu flower
 <img src="https://huggingface.co/spaces/clip-italian/clip-italian-demo/raw/main/static/img/fiore_blu.png" alt="drawing" width="600"/>
 
 ### Counting
-What about "one cat"
+What about "one cat"?
 <img src="https://huggingface.co/spaces/clip-italian/clip-italian-demo/raw/main/static/img/gatto.png" alt="drawing" width="600"/>
 
 And what about "two cats"?

text2image.py

@@ -103,7 +103,11 @@ def app():
     
         ### 👋 Ciao!
 
-        Here you can search for images in the Unsplash 25k Photos dataset.
+        Here you can search for images in the Unsplash 25k Photos dataset and the Conceptual Caption dataset.
+        You will see most queries make sense. When you see errors, there might be two possibilities: the model is answering
+        in a wrong way or the image you are looking for and the model is giving you the best answer it can get.
+        
+        
         
         🤌 Italian mode on! 🤌
 
@@ -129,7 +133,7 @@ def app():
         )
     with col4:
         st.button(
-            "Un fiore blu", on_click=update_query, kwargs=dict(value="Un fiore blu")
+            "Un gatto sopra una sedia", on_click=update_query, kwargs=dict(value="Un gatto sopra una sedia")
         )
 
     col1, col2 = st.beta_columns([3, 1])
@@ -163,7 +167,7 @@ def app():
                 raise ValueError()
 
             image_paths = utils.find_image(
-                query, model, dataset, tokenizer, image_features, 2, dataset_name
+                query, model, dataset, tokenizer, image_features, 1, dataset_name
             )
 
         st.image(image_paths)