orrinin commited on
Commit
d6d54c5
·
verified ·
1 Parent(s): 44998dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -4
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 = embedding_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