Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,28 @@ import torch
|
|
3 |
import torch.nn.functional as F
|
4 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
5 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
# Load model and tokenizer from Hugging Face Hub
|
7 |
@st.cache_resource
|
8 |
def load_model_and_tokenizer():
|
@@ -14,16 +36,21 @@ def load_model_and_tokenizer():
|
|
14 |
# Load resources
|
15 |
model, tokenizer = load_model_and_tokenizer()
|
16 |
|
17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
st.write(
|
19 |
"Въведете една или повече заявки (всеки на нов ред) или качете `.txt` файл, в който "
|
20 |
"всяка заявка е на отделен ред без допълнителни параметри."
|
21 |
)
|
22 |
|
23 |
# Display author info
|
24 |
-
st.markdown(
|
25 |
-
"### Моделът е създаден от [DEJAN AI](https://dejan.ai)"
|
26 |
-
)
|
27 |
|
28 |
# Текстово поле за въвеждане на заявки
|
29 |
queries_input = st.text_area("Въведете вашите заявки (по една на ред):")
|
|
|
3 |
import torch.nn.functional as F
|
4 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
5 |
|
6 |
+
# Hide Streamlit's menu and footer
|
7 |
+
st.markdown("""
|
8 |
+
<style>
|
9 |
+
#MainMenu {visibility: hidden;}
|
10 |
+
footer {visibility: hidden;}
|
11 |
+
header {visibility: hidden;}
|
12 |
+
.block-container {
|
13 |
+
padding-top: 1rem;
|
14 |
+
}
|
15 |
+
.st-emotion-cache-16idsys {
|
16 |
+
display: flex;
|
17 |
+
justify-content: space-between;
|
18 |
+
align-items: center;
|
19 |
+
}
|
20 |
+
.logo-container {
|
21 |
+
position: absolute;
|
22 |
+
top: 10px;
|
23 |
+
right: 10px;
|
24 |
+
}
|
25 |
+
</style>
|
26 |
+
""", unsafe_allow_html=True)
|
27 |
+
|
28 |
# Load model and tokenizer from Hugging Face Hub
|
29 |
@st.cache_resource
|
30 |
def load_model_and_tokenizer():
|
|
|
36 |
# Load resources
|
37 |
model, tokenizer = load_model_and_tokenizer()
|
38 |
|
39 |
+
# Page layout with logo
|
40 |
+
st.markdown("""
|
41 |
+
<div style="display: flex; justify-content: space-between; align-items: center;">
|
42 |
+
<h1>Класификация на намерения за търсене (Български)</h1>
|
43 |
+
<img src="https://huggingface.co/spaces/dejanseo/bulgarian-search-query-intent-classifier/resolve/main/dejan-300x103.png" width="300">
|
44 |
+
</div>
|
45 |
+
""", unsafe_allow_html=True)
|
46 |
+
|
47 |
st.write(
|
48 |
"Въведете една или повече заявки (всеки на нов ред) или качете `.txt` файл, в който "
|
49 |
"всяка заявка е на отделен ред без допълнителни параметри."
|
50 |
)
|
51 |
|
52 |
# Display author info
|
53 |
+
st.markdown("### Моделът е създаден от [DEJAN AI](https://dejan.ai)")
|
|
|
|
|
54 |
|
55 |
# Текстово поле за въвеждане на заявки
|
56 |
queries_input = st.text_area("Въведете вашите заявки (по една на ред):")
|