Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ with col1:
|
|
| 22 |
with col2:
|
| 23 |
model_type = st.selectbox("Какой моделью пользуемся?\n ", options=['rubert-tiny2', 'msmarco-MiniLM-L-12-v3'])
|
| 24 |
with col3:
|
| 25 |
-
genres_list = st.multiselect("Какого
|
| 26 |
|
| 27 |
if model_type == 'rubert-tiny2':
|
| 28 |
model = SentenceTransformer('cointegrated/rubert-tiny2')
|
|
|
|
| 22 |
with col2:
|
| 23 |
model_type = st.selectbox("Какой моделью пользуемся?\n ", options=['rubert-tiny2', 'msmarco-MiniLM-L-12-v3'])
|
| 24 |
with col3:
|
| 25 |
+
genres_list = st.multiselect("Какого жанра фильм?\n ", options=all_genres)
|
| 26 |
|
| 27 |
if model_type == 'rubert-tiny2':
|
| 28 |
model = SentenceTransformer('cointegrated/rubert-tiny2')
|