sritang commited on
Commit
4a4efca
·
1 Parent(s): 411aa02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("gfhayworth/hack_policy_embed_sri", split='train')
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