Tonic commited on
Commit
ecb8e22
·
1 Parent(s): 79ad113

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,7 +4,8 @@ import gradio as gr
4
  from langchain.embeddings import CohereEmbeddings
5
  from langchain.document_loaders import UnstructuredFileLoader
6
  from langchain.vectorstores import Weaviate
7
- from langchain.chain_types import OpenAI, RetrievalQA
 
8
  import os
9
  import urllib.request
10
  import ssl
@@ -25,7 +26,7 @@ client = weaviate.Client(url=weaviate_url, auth_client_secret=auth_config,
25
  additional_headers={"X-Cohere-Api-Key": cohere_api_key})
26
 
27
  # Initialize vectorstore
28
- vectorstore = Weaviate(client, index_name="Articles", text_key="text")
29
  vectorstore._query_attrs = ["text", "title", "url", "views", "lang", "_additional {distance}"]
30
  vectorstore.embedding = CohereEmbeddings(model="embed-multilingual-v2.0", cohere_api_key=cohere_api_key)
31
 
 
4
  from langchain.embeddings import CohereEmbeddings
5
  from langchain.document_loaders import UnstructuredFileLoader
6
  from langchain.vectorstores import Weaviate
7
+ from langchain.llms import OpenAI
8
+ from langchain.chains import RetrievalQA
9
  import os
10
  import urllib.request
11
  import ssl
 
26
  additional_headers={"X-Cohere-Api-Key": cohere_api_key})
27
 
28
  # Initialize vectorstore
29
+ vectorstore = Weaviate(client, index_name="HereChat", text_key="text")
30
  vectorstore._query_attrs = ["text", "title", "url", "views", "lang", "_additional {distance}"]
31
  vectorstore.embedding = CohereEmbeddings(model="embed-multilingual-v2.0", cohere_api_key=cohere_api_key)
32