Ronoh4
commited on
Commit
·
d930f2c
1
Parent(s):
9535c62
Added app file
Browse files
app.py
CHANGED
|
@@ -36,12 +36,10 @@ embed_model = NVIDIAEmbedding(
|
|
| 36 |
|
| 37 |
|
| 38 |
reranker = NVIDIARerank(
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
truncate="NONE"
|
| 42 |
)
|
| 43 |
|
| 44 |
-
|
| 45 |
# Set the NVIDIA models globally
|
| 46 |
Settings.embed_model = embed_model
|
| 47 |
Settings.llm = client
|
|
|
|
| 36 |
|
| 37 |
|
| 38 |
reranker = NVIDIARerank(
|
| 39 |
+
model="nvidia/nv-rerankqa-mistral-4b-v3",
|
| 40 |
+
api_key=nvidia_api_key,
|
|
|
|
| 41 |
)
|
| 42 |
|
|
|
|
| 43 |
# Set the NVIDIA models globally
|
| 44 |
Settings.embed_model = embed_model
|
| 45 |
Settings.llm = client
|