add markdown descriptions

app.py

@@ -118,17 +118,31 @@ def predict_subset(model_id, token):
     from toolz import frequencies
 
     df = pd.DataFrame(
-        {"labels": frequencies(labels).keys(), "freqs": frequencies(labels).values()}
+        {
+            "labels": frequencies(labels).keys(),
+            "freqs": frequencies(labels).values(),
+        }
     )
     return gallery, df
 
 
 with gr.Blocks() as demo:
     with gr.Tab("Random image gallery"):
+        gr.Markdown(
+            """## Random image gallery
+        This is a random image gallery. 
+        You can refresh the images by clicking the refresh button."""
+        )
         button = gr.Button("Refresh")
         gallery = gr.Gallery().style(grid=9, height="1400")
         button.click(return_random_sample, [], [gallery])
     with gr.Tab("image search"):
+        gr.Markdown(
+            """## Image search
+        This is an image search.    
+        You can search for images by entering a search term and clicking the search button.
+        You can also change the number of images to be returned."""
+        )
         text = gr.Textbox(label="Search for images")
         k = gr.Slider(minimum=3, maximum=18, step=1)
         button = gr.Button("search")
@@ -143,9 +157,14 @@ with gr.Blocks() as demo:
     #     csv_file = gr.File("label_studio.csv")
     #     button.click(dataset.save_to_disk, [], [csv_file])
     with gr.Tab("predict"):
+        gr.Markdown(
+            """## Image classification model tester
+        You can use this to test out [image classification models](https://huggingface.co/models?pipeline_tag=image-classification) on the Hugging Face Hub. """
+        )
         token = gr.Textbox(label="token", type="password")
         model_id = gr.Textbox(label="model_id")
         button = gr.Button("predict")
+        gr.Markdown("## Results")
         plot = gr.BarPlot(x="labels", y="freqs", width=600, height=400, vertical=False)
         gallery = gr.Gallery()
         button.click(predict_subset, [model_id, token], [gallery, plot])