Spaces:
Runtime error
Runtime error
Ashmi Banerjee
commited on
Commit
Β·
c59035e
1
Parent(s):
420fa8a
updates with gemini
Browse files- .gitignore +2 -2
- build_rag.py +1 -1
- models/gemini.py +1 -1
- {setups β setup}/db_setup.py +0 -0
- {setups β setup}/vertex_ai_setup.py +0 -0
.gitignore
CHANGED
|
@@ -180,5 +180,5 @@ gradio_cached_examples/
|
|
| 180 |
__pycache__/
|
| 181 |
gradio_cached_examples/
|
| 182 |
models/__pycache__/
|
| 183 |
-
|
| 184 |
-
|
|
|
|
| 180 |
__pycache__/
|
| 181 |
gradio_cached_examples/
|
| 182 |
models/__pycache__/
|
| 183 |
+
setup/__pycache__/
|
| 184 |
+
setup/gcp-default-creds.json
|
build_rag.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
from sentence_transformers import SentenceTransformer
|
| 2 |
-
from
|
| 3 |
|
| 4 |
|
| 5 |
def get_embedding(text: str) -> list[float]:
|
|
|
|
| 1 |
from sentence_transformers import SentenceTransformer
|
| 2 |
+
from setup.db_setup import get_mongo_client, get_mongo_url
|
| 3 |
|
| 4 |
|
| 5 |
def get_embedding(text: str) -> list[float]:
|
models/gemini.py
CHANGED
|
@@ -2,7 +2,7 @@ from typing import Optional
|
|
| 2 |
import sys
|
| 3 |
|
| 4 |
sys.path.append("../")
|
| 5 |
-
from
|
| 6 |
from vertexai.preview.generative_models import GenerativeModel
|
| 7 |
|
| 8 |
|
|
|
|
| 2 |
import sys
|
| 3 |
|
| 4 |
sys.path.append("../")
|
| 5 |
+
from setup.vertex_ai_setup import initialize_vertexai_params
|
| 6 |
from vertexai.preview.generative_models import GenerativeModel
|
| 7 |
|
| 8 |
|
{setups β setup}/db_setup.py
RENAMED
|
File without changes
|
{setups β setup}/vertex_ai_setup.py
RENAMED
|
File without changes
|