HaggiVaggi commited on
Commit
58f8537
1 Parent(s): 6d92d89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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('data_with_adds.csv')
58
- embeddings_array = np.load('embeddings.npy')
59
- index = faiss.read_index('desc_faiss_index.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")