NCTCMumbai
commited on
Commit
•
d1079ea
1
Parent(s):
a66eff8
Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ bert_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
|
|
50 |
def get_bert_embeddings(corpus, model):
|
51 |
return model.encode(corpus)
|
52 |
|
53 |
-
def load_embeddings(filename="
|
54 |
embeddings = np.load(filename)
|
55 |
return embeddings
|
56 |
|
@@ -107,7 +107,7 @@ def search_and_explain(query):
|
|
107 |
# Replace the following line with actual loading of the DataFrame
|
108 |
global df
|
109 |
|
110 |
-
df = pd.read_csv('
|
111 |
|
112 |
corpus = df['Concat Description'].tolist()
|
113 |
|
|
|
50 |
def get_bert_embeddings(corpus, model):
|
51 |
return model.encode(corpus)
|
52 |
|
53 |
+
def load_embeddings(filename="corpus_embeddings.npy"):
|
54 |
embeddings = np.load(filename)
|
55 |
return embeddings
|
56 |
|
|
|
107 |
# Replace the following line with actual loading of the DataFrame
|
108 |
global df
|
109 |
|
110 |
+
df = pd.read_csv('CTH_Description (2).csv', on_bad_lines = 'skip' ) # Load your data
|
111 |
|
112 |
corpus = df['Concat Description'].tolist()
|
113 |
|