Update app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ def generate_memes(prompt):
|
|
| 37 |
print(f"Error loading image {url}: {e}")
|
| 38 |
return images
|
| 39 |
|
| 40 |
-
# UI
|
| 41 |
input_textbox = gr.Textbox(lines=1, label="Search something cool")
|
| 42 |
output_gallery = gr.Gallery(label="Retrieved Memes", columns=3, rows=2, height="auto")
|
| 43 |
|
|
@@ -45,6 +45,7 @@ title = "Semantic Search for Memes"
|
|
| 45 |
description = "Search memes from a dataset of ~6k memes using semantic similarity. [GitHub Repo](https://github.com/bhavya-giri/retrieving-memes)"
|
| 46 |
examples = ["Get Shreked", "Going Crazy", "Spiderman is my teacher"]
|
| 47 |
|
|
|
|
| 48 |
iface = gr.Interface(
|
| 49 |
fn=generate_memes,
|
| 50 |
inputs=input_textbox,
|
|
@@ -52,8 +53,7 @@ iface = gr.Interface(
|
|
| 52 |
examples=examples,
|
| 53 |
cache_examples=True,
|
| 54 |
title=title,
|
| 55 |
-
description=description
|
| 56 |
-
enable_queue=True
|
| 57 |
)
|
| 58 |
|
| 59 |
iface.launch()
|
|
|
|
| 37 |
print(f"Error loading image {url}: {e}")
|
| 38 |
return images
|
| 39 |
|
| 40 |
+
# UI components
|
| 41 |
input_textbox = gr.Textbox(lines=1, label="Search something cool")
|
| 42 |
output_gallery = gr.Gallery(label="Retrieved Memes", columns=3, rows=2, height="auto")
|
| 43 |
|
|
|
|
| 45 |
description = "Search memes from a dataset of ~6k memes using semantic similarity. [GitHub Repo](https://github.com/bhavya-giri/retrieving-memes)"
|
| 46 |
examples = ["Get Shreked", "Going Crazy", "Spiderman is my teacher"]
|
| 47 |
|
| 48 |
+
# Interface
|
| 49 |
iface = gr.Interface(
|
| 50 |
fn=generate_memes,
|
| 51 |
inputs=input_textbox,
|
|
|
|
| 53 |
examples=examples,
|
| 54 |
cache_examples=True,
|
| 55 |
title=title,
|
| 56 |
+
description=description
|
|
|
|
| 57 |
)
|
| 58 |
|
| 59 |
iface.launch()
|