Update app.py
Browse files
app.py
CHANGED
@@ -28,14 +28,22 @@ llm = Cohere(
|
|
28 |
embedding_model = CohereEmbedding(
|
29 |
api_key=api_key,
|
30 |
model_name="embed-multilingual-v3.0",
|
31 |
-
input_type="search_document",
|
32 |
-
|
33 |
|
|
|
|
|
|
|
|
|
34 |
|
35 |
|
36 |
# Set Global settings
|
37 |
Settings.llm = llm
|
38 |
-
Settings.embed_model
|
|
|
|
|
|
|
|
|
39 |
|
40 |
db_path=""
|
41 |
|
@@ -118,7 +126,7 @@ def infer(message:str, history: list):
|
|
118 |
|
119 |
# Get the index from the vector store
|
120 |
index = VectorStoreIndex.from_vector_store(
|
121 |
-
vector_store
|
122 |
)
|
123 |
|
124 |
|
|
|
28 |
embedding_model = CohereEmbedding(
|
29 |
api_key=api_key,
|
30 |
model_name="embed-multilingual-v3.0",
|
31 |
+
input_type="search_document",
|
32 |
+
embedding_type="int8",)
|
33 |
|
34 |
+
query_model = CohereEmbedding(
|
35 |
+
api_key=api_key,
|
36 |
+
model_name="embed-multilingual-v3.0",
|
37 |
+
input_type="search_query",)
|
38 |
|
39 |
|
40 |
# Set Global settings
|
41 |
Settings.llm = llm
|
42 |
+
Settings.embed_model=embedding_model
|
43 |
+
# set context window
|
44 |
+
Settings.context_window = 4096
|
45 |
+
# set number of output tokens
|
46 |
+
Settings.num_output = 256
|
47 |
|
48 |
db_path=""
|
49 |
|
|
|
126 |
|
127 |
# Get the index from the vector store
|
128 |
index = VectorStoreIndex.from_vector_store(
|
129 |
+
vector_store,
|
130 |
)
|
131 |
|
132 |
|