Update app.py
Browse files
app.py
CHANGED
|
@@ -34,10 +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 |
-
print(df)
|
|
|
|
| 38 |
df['embeddings'] = df['text'].apply(lambda x: embedding_model.encode(x))
|
| 39 |
print(embeddings)
|
| 40 |
-
print(df)
|
| 41 |
dataset = Dataset.from_pandas(df)
|
| 42 |
print(dataset[1])
|
| 43 |
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.iloc[[1]])
|
| 38 |
+
|
| 39 |
df['embeddings'] = df['text'].apply(lambda x: embedding_model.encode(x))
|
| 40 |
print(embeddings)
|
| 41 |
+
print(df.iloc[[1]])
|
| 42 |
dataset = Dataset.from_pandas(df)
|
| 43 |
print(dataset[1])
|
| 44 |
print(dataset[2])
|