Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
|
|
| 39 |
dataset = load_dataset("sritang/hack_df", split='train')
|
| 40 |
mypassages = list(dataset.to_pandas()['paragraph'])
|
| 41 |
|
| 42 |
-
dataset_embed = load_dataset("
|
| 43 |
dataset_embed_pd = dataset_embed.to_pandas()
|
| 44 |
mycorpus_embeddings = torch_tensor(dataset_embed_pd.values)
|
| 45 |
|
|
|
|
| 39 |
dataset = load_dataset("sritang/hack_df", split='train')
|
| 40 |
mypassages = list(dataset.to_pandas()['paragraph'])
|
| 41 |
|
| 42 |
+
dataset_embed = load_dataset("sritang/hack_policy_embed_sri", split='train')
|
| 43 |
dataset_embed_pd = dataset_embed.to_pandas()
|
| 44 |
mycorpus_embeddings = torch_tensor(dataset_embed_pd.values)
|
| 45 |
|