vjain commited on
Commit
5cd4f42
1 Parent(s): f5e51e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,10 +4,10 @@ import pandas as pd
4
  import numpy as np
5
  import csv
6
  import os
 
 
7
  from datasets import load_dataset
8
  openai.api_key= os.environ.get("openai.api_key")
9
- from openai.embeddings_utils import get_embedding
10
- from openai.embeddings_utils import cosine_similarity
11
  import requests
12
  model_id = "sentence-transformers/all-MiniLM-L6-v2"
13
  import json
@@ -22,8 +22,8 @@ def generate_embeddings(texts, model_id, hf_token):
22
  embeddings = response.json()
23
  return embeddings
24
  AP_Bio = load_dataset('vjain/biology_AP_embeddings')
25
- df1 = pd.DataFrame(AP_Bio['train'])
26
- df1["similarity"] = 0
27
 
28
  AP_Physics = load_dataset('vjain/AP_physics_embeddings')
29
  df2 = pd.DataFrame(AP_Physics['train'])
 
4
  import numpy as np
5
  import csv
6
  import os
7
+ from sentence_transformers.util import semantic_search
8
+
9
  from datasets import load_dataset
10
  openai.api_key= os.environ.get("openai.api_key")
 
 
11
  import requests
12
  model_id = "sentence-transformers/all-MiniLM-L6-v2"
13
  import json
 
22
  embeddings = response.json()
23
  return embeddings
24
  AP_Bio = load_dataset('vjain/biology_AP_embeddings')
25
+ dataset_embeddings1 = torch.from_numpy(AP_bio["train"].to_pandas().to_numpy()).to(torch.float)
26
+
27
 
28
  AP_Physics = load_dataset('vjain/AP_physics_embeddings')
29
  df2 = pd.DataFrame(AP_Physics['train'])