Update app.py
Browse files
app.py
CHANGED
|
@@ -34,11 +34,11 @@ embedding_model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
|
|
| 34 |
#all-MiniLM-L6-v2, BAAI/bge-base-en-v1.5,infgrad/stella-base-en-v2, BAAI/bge-large-en-v1.5 working with default dimensions
|
| 35 |
|
| 36 |
df = pd.DataFrame(dataset)
|
| 37 |
-
|
| 38 |
embeddings = embedding_model.encode(dataset["text"])
|
| 39 |
print(embeddings)
|
| 40 |
df['embeddings'] = embeddings
|
| 41 |
-
|
| 42 |
dataset = Dataset.from_pandas(df)
|
| 43 |
print(dataset[1])
|
| 44 |
print(dataset[2])
|
|
|
|
| 34 |
#all-MiniLM-L6-v2, BAAI/bge-base-en-v1.5,infgrad/stella-base-en-v2, BAAI/bge-large-en-v1.5 working with default dimensions
|
| 35 |
|
| 36 |
df = pd.DataFrame(dataset)
|
| 37 |
+
print(df)
|
| 38 |
embeddings = embedding_model.encode(dataset["text"])
|
| 39 |
print(embeddings)
|
| 40 |
df['embeddings'] = embeddings
|
| 41 |
+
print(df)
|
| 42 |
dataset = Dataset.from_pandas(df)
|
| 43 |
print(dataset[1])
|
| 44 |
print(dataset[2])
|