Cyanido commited on
Commit
c1125a1
·
verified ·
1 Parent(s): b1fafc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -1,10 +1,10 @@
1
  # Initialize a retriever using Qdrant and SentenceTransformer embeddings
2
- from langchain_community.vectorstores import Qdrant
3
- from langchain_community.embeddings import SentenceTransformerEmbeddings
4
- from kaggle_secrets import UserSecretsClient
5
  from qdrant_client import QdrantClient
6
  import pandas as pd
7
- import gradio as gd
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()