Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,8 @@ from sentence_transformers.util import cos_sim
|
|
| 6 |
from sentence_transformers import SentenceTransformer
|
| 7 |
import gradio as gr
|
| 8 |
#%%
|
| 9 |
-
etalon = pd.read_csv("
|
| 10 |
-
df = pd.read_csv("
|
| 11 |
df = df[df['is_match'] == 1]
|
| 12 |
|
| 13 |
model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2').to("cuda")
|
|
@@ -16,7 +16,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
| 16 |
|
| 17 |
unique_complaints = df['Жалобы'].values.tolist()
|
| 18 |
|
| 19 |
-
with open("
|
| 20 |
unique_complaints_embeddings_st = np.load(f)
|
| 21 |
|
| 22 |
def get_recommend(user_input,
|
|
|
|
| 6 |
from sentence_transformers import SentenceTransformer
|
| 7 |
import gradio as gr
|
| 8 |
#%%
|
| 9 |
+
etalon = pd.read_csv("etalon_prod.csv")
|
| 10 |
+
df = pd.read_csv("preprocessed_train_classify_rec_spec_filtered_by_etalon.csv")
|
| 11 |
df = df[df['is_match'] == 1]
|
| 12 |
|
| 13 |
model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2').to("cuda")
|
|
|
|
| 16 |
|
| 17 |
unique_complaints = df['Жалобы'].values.tolist()
|
| 18 |
|
| 19 |
+
with open("embeddings.npy", 'rb') as f:
|
| 20 |
unique_complaints_embeddings_st = np.load(f)
|
| 21 |
|
| 22 |
def get_recommend(user_input,
|