Update app.py

app.py

@@ -4,10 +4,10 @@ import pandas as pd
 import numpy as np
 import csv
 import os
+from sentence_transformers.util import semantic_search
+
 from datasets import load_dataset
 openai.api_key= os.environ.get("openai.api_key")
-from openai.embeddings_utils import get_embedding
-from openai.embeddings_utils import cosine_similarity
 import requests
 model_id = "sentence-transformers/all-MiniLM-L6-v2"
 import json
@@ -22,8 +22,8 @@ def generate_embeddings(texts, model_id, hf_token):
     embeddings = response.json()
     return embeddings
 AP_Bio = load_dataset('vjain/biology_AP_embeddings')
-df1 = pd.DataFrame(AP_Bio['train'])
-df1["similarity"] = 0
+dataset_embeddings1 = torch.from_numpy(AP_bio["train"].to_pandas().to_numpy()).to(torch.float)
+
 
 AP_Physics = load_dataset('vjain/AP_physics_embeddings')
 df2 = pd.DataFrame(AP_Physics['train'])