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()
|