Spaces:
Sleeping
Sleeping
Prathmesh48
commited on
Commit
•
d54e971
1
Parent(s):
b70cf98
Update embedding.py
Browse files- embedding.py +2 -2
embedding.py
CHANGED
@@ -28,8 +28,8 @@ vision1 = ChatGoogleGenerativeAI(model="gemini-1.5-flash",google_api_key='AIzaSy
|
|
28 |
vision2 = ChatGoogleGenerativeAI(model="gemini-1.5-flash",google_api_key='AIzaSyBzbZQBffHFK3N-gWnhDDNbQ9yZnZtaS2E',temperature = 0.1)
|
29 |
vision3 = ChatGoogleGenerativeAI(model="gemini-1.5-flash",google_api_key='AIzaSyBNN4VDMAOB2gSZha6HjsTuH71PVV69FLM',temperature = 0.1)
|
30 |
|
31 |
-
tokenizer = AutoTokenizer.from_pretrained('
|
32 |
-
model = AutoModel.from_pretrained('
|
33 |
model.to('cpu') # Ensure the model is on the CPU
|
34 |
|
35 |
|
|
|
28 |
vision2 = ChatGoogleGenerativeAI(model="gemini-1.5-flash",google_api_key='AIzaSyBzbZQBffHFK3N-gWnhDDNbQ9yZnZtaS2E',temperature = 0.1)
|
29 |
vision3 = ChatGoogleGenerativeAI(model="gemini-1.5-flash",google_api_key='AIzaSyBNN4VDMAOB2gSZha6HjsTuH71PVV69FLM',temperature = 0.1)
|
30 |
|
31 |
+
tokenizer = AutoTokenizer.from_pretrained('Alibaba-NLP/gte-base-en-v1.5',trust_remote_code = True)
|
32 |
+
model = AutoModel.from_pretrained('Alibaba-NLP/gte-base-en-v1.5',trust_remote_code = True)
|
33 |
model.to('cpu') # Ensure the model is on the CPU
|
34 |
|
35 |
|