Update to description

app.py

@@ -79,21 +79,21 @@ def find_similar_images(text):
     
     return top_images
 
-# Description of classes
-description_text = """
-This application classifies images into one of four classes:
-- **Deer**: Images containing deer or related visuals.
-- **Rabbit**: Images containing rabbits or similar visuals.
-- **Dog**: Images containing various dog breeds.
-- **Shark**: Images with sharks or related visuals.
-
-Enter a description, and the model will return the top 4 most similar images.
-"""
-
-# Interface setup
+
+
 text_input = gr.Textbox(label="Input text", placeholder="Enter the images description")
 imgs_output = gr.Gallery(label="Top 4 most similar images")
 
-intf = gr.Interface(fn=find_similar_images, inputs=text_input, outputs=imgs_output, description=description_text)
-
-intf.launch(share=True)
+intf = gr.Interface(
+    fn=find_similar_images,
+    inputs=gr.Textbox(label="Input Text", placeholder="Enter a description"),
+    outputs=gr.Gallery(label="Top 4 Similar Images"),
+)
+extra_text = gr.Markdown("""
+The dataset contains images of dogs, rabbits, sharks, and deer.
+Displaying the images might take a couple of seconds.
+""")
+with gr.Blocks() as app:
+    intf.render()
+    extra_text.render()
+app.launch(share=True)