Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ df['embeddings'] = df['text'].apply(lambda x: embedding_model.encode(x))
|
|
| 31 |
# add_embeddings as a new column
|
| 32 |
|
| 33 |
print("check1a")
|
| 34 |
-
print(df.iloc[[1]])
|
| 35 |
dataset = Dataset.from_pandas(df)
|
| 36 |
print("check1b")
|
| 37 |
|
|
@@ -43,7 +43,7 @@ embedding_dim = embedding_model.get_sentence_embedding_dimension()
|
|
| 43 |
# Returns dimensions of embedidng
|
| 44 |
data = dataset
|
| 45 |
|
| 46 |
-
|
| 47 |
d = 384 # vectors dimension
|
| 48 |
m = 32 # hnsw parameter. Higher is more accurate but takes more time to index (default is 32, 128 should be ok)
|
| 49 |
#index = faiss.IndexHNSWFlat(d, m)
|
|
|
|
| 31 |
# add_embeddings as a new column
|
| 32 |
|
| 33 |
print("check1a")
|
| 34 |
+
#print(df.iloc[[1]])
|
| 35 |
dataset = Dataset.from_pandas(df)
|
| 36 |
print("check1b")
|
| 37 |
|
|
|
|
| 43 |
# Returns dimensions of embedidng
|
| 44 |
data = dataset
|
| 45 |
|
| 46 |
+
print(embedding_dim)
|
| 47 |
d = 384 # vectors dimension
|
| 48 |
m = 32 # hnsw parameter. Higher is more accurate but takes more time to index (default is 32, 128 should be ok)
|
| 49 |
#index = faiss.IndexHNSWFlat(d, m)
|