Spaces:
Sleeping
Sleeping
pbrenotvinciguerra
commited on
Commit
•
c4b67c7
1
Parent(s):
4f265da
back to mistral 7b model
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ from llama_index.query_engine import RetrieverQueryEngine
|
|
10 |
from llama_index.llms import Anyscale
|
11 |
|
12 |
# Define the inference model
|
13 |
-
|
14 |
-
llm = Anyscale(model="HuggingFaceH4/zephyr-7b-beta", api_key=os.getenv("ANYSCALE_API_KEY"))
|
15 |
# Define the embedding model used to embed the query.
|
16 |
# query_embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5")
|
17 |
embed_model = VoyageEmbedding(model_name="voyage-01", voyage_api_key=os.getenv("VOYAGE_API_KEY"))
|
|
|
10 |
from llama_index.llms import Anyscale
|
11 |
|
12 |
# Define the inference model
|
13 |
+
llm = Anyscale(model="mistralai/Mistral-7B-Instruct-v0.1", api_key=os.getenv("ANYSCALE_API_KEY"))
|
14 |
+
#llm = Anyscale(model="HuggingFaceH4/zephyr-7b-beta", api_key=os.getenv("ANYSCALE_API_KEY"))
|
15 |
# Define the embedding model used to embed the query.
|
16 |
# query_embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5")
|
17 |
embed_model = VoyageEmbedding(model_name="voyage-01", voyage_api_key=os.getenv("VOYAGE_API_KEY"))
|