Spaces:
Sleeping
Sleeping
Commit
•
58f8537
1
Parent(s):
6d92d89
Update app.py
Browse files
app.py
CHANGED
@@ -49,14 +49,14 @@ if page == "Главная":
|
|
49 |
st.write("Случайные 10 фильмов")
|
50 |
st.write(random_rows)
|
51 |
|
52 |
-
if page == "
|
53 |
# Загрузка предварительно обученной модели ruBERT
|
54 |
tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
|
55 |
model = AutoModel.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
|
56 |
|
57 |
-
df2 = pd.read_csv('
|
58 |
-
embeddings_array = np.load('
|
59 |
-
index = faiss.read_index('
|
60 |
|
61 |
def encode_description(description):
|
62 |
tokens = tokenizer(description, return_tensors="pt")
|
|
|
49 |
st.write("Случайные 10 фильмов")
|
50 |
st.write(random_rows)
|
51 |
|
52 |
+
if page == "Подбор фильмов по описанию":
|
53 |
# Загрузка предварительно обученной модели ruBERT
|
54 |
tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
|
55 |
model = AutoModel.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
|
56 |
|
57 |
+
df2 = pd.read_csv('fina_data.csv')
|
58 |
+
embeddings_array = np.load('embeddings_final.npy')
|
59 |
+
index = faiss.read_index('desc_faiss_index_final.index')
|
60 |
|
61 |
def encode_description(description):
|
62 |
tokens = tokenizer(description, return_tensors="pt")
|