watagassy commited on
Commit
a4531c0
1 Parent(s): f54a3d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -12,13 +12,13 @@ qdrant_url = os.environ['QDRANT_URL']
12
  qdrant_api_key = os.environ['QDRANT_API_KEY']
13
 
14
  def process_text(input_text, top_k):
15
- # embeddings = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-base")
16
- embeddings = HuggingFaceEmbeddings(model_name="nlp-waseda/roberta-large-japanese")
17
  client = QdrantClient(
18
  url=qdrant_url, api_key=qdrant_api_key, prefer_grpc=False,
19
  )
20
- # db = Qdrant(client=client, embeddings=embeddings, collection_name="qa_data_without_cardname")
21
- db = Qdrant(client=client, embeddings=embeddings, collection_name="qa_data_without_cardname_rlj")
22
  query = input_text
23
 
24
  all_answers = []
 
12
  qdrant_api_key = os.environ['QDRANT_API_KEY']
13
 
14
  def process_text(input_text, top_k):
15
+ embeddings = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-base")
16
+ # embeddings = HuggingFaceEmbeddings(model_name="nlp-waseda/roberta-large-japanese")
17
  client = QdrantClient(
18
  url=qdrant_url, api_key=qdrant_api_key, prefer_grpc=False,
19
  )
20
+ db = Qdrant(client=client, embeddings=embeddings, collection_name="qa_data_without_cardname")
21
+ # db = Qdrant(client=client, embeddings=embeddings, collection_name="qa_data_without_cardname_rlj")
22
  query = input_text
23
 
24
  all_answers = []