Spaces:
Running
Running
AFischer1985
commited on
Update run.py
Browse files
run.py
CHANGED
@@ -40,7 +40,7 @@ from chromadb.utils import embedding_functions
|
|
40 |
default_ef = embedding_functions.DefaultEmbeddingFunction()
|
41 |
#sentence_transformer_ef = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer")
|
42 |
#instructor_ef = embedding_functions.InstructorEmbeddingFunction(model_name="hkunlp/instructor-large", device="cuda")
|
43 |
-
embeddingModel = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer"
|
44 |
|
45 |
print(str(client.list_collections()))
|
46 |
|
|
|
40 |
default_ef = embedding_functions.DefaultEmbeddingFunction()
|
41 |
#sentence_transformer_ef = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer")
|
42 |
#instructor_ef = embedding_functions.InstructorEmbeddingFunction(model_name="hkunlp/instructor-large", device="cuda")
|
43 |
+
embeddingModel = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer") #, device="cuda")
|
44 |
|
45 |
print(str(client.list_collections()))
|
46 |
|