Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,10 +1,10 @@
|
|
| 1 |
# Initialize a retriever using Qdrant and SentenceTransformer embeddings
|
| 2 |
-
from
|
| 3 |
-
from
|
| 4 |
-
from
|
| 5 |
from qdrant_client import QdrantClient
|
| 6 |
import pandas as pd
|
| 7 |
-
import gradio as
|
| 8 |
|
| 9 |
|
| 10 |
embeddings = SentenceTransformerEmbeddings(model_name='sentence-transformers/clip-ViT-B-32')
|
|
@@ -13,7 +13,6 @@ def get_results(search_results):
|
|
| 13 |
filtered_img_ids = [doc.metadata.get("image_id") for doc in search_results]
|
| 14 |
return filtered_img_ids
|
| 15 |
|
| 16 |
-
user_secrets = UserSecretsClient()
|
| 17 |
vector_db_key = user_secrets.get_secret("vector_db_key")
|
| 18 |
|
| 19 |
client = QdrantClient(
|
|
@@ -39,4 +38,4 @@ def get_link(query):
|
|
| 39 |
return links
|
| 40 |
|
| 41 |
# print(get_link("black shirt for men"))
|
| 42 |
-
gr.Interface(fn = get_link, inputs = 'textbox', outputs = 'textbox').launch()
|
|
|
|
| 1 |
# Initialize a retriever using Qdrant and SentenceTransformer embeddings
|
| 2 |
+
from langchain.vectorstores import Qdrant
|
| 3 |
+
from langchain.retrievers import EnsembleRetriever
|
| 4 |
+
from langchain.embeddings import SentenceTransformerEmbeddings
|
| 5 |
from qdrant_client import QdrantClient
|
| 6 |
import pandas as pd
|
| 7 |
+
import gradio as gr
|
| 8 |
|
| 9 |
|
| 10 |
embeddings = SentenceTransformerEmbeddings(model_name='sentence-transformers/clip-ViT-B-32')
|
|
|
|
| 13 |
filtered_img_ids = [doc.metadata.get("image_id") for doc in search_results]
|
| 14 |
return filtered_img_ids
|
| 15 |
|
|
|
|
| 16 |
vector_db_key = user_secrets.get_secret("vector_db_key")
|
| 17 |
|
| 18 |
client = QdrantClient(
|
|
|
|
| 38 |
return links
|
| 39 |
|
| 40 |
# print(get_link("black shirt for men"))
|
| 41 |
+
gr.Interface(fn = get_link, inputs = 'textbox', outputs = 'textbox').launch()
|