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