dejanseo commited on
Commit
a8cfa00
·
verified ·
1 Parent(s): e0c54fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -4
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
- st.title("Класификация на намерения за търсене (Български)")
 
 
 
 
 
 
 
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("Въведете вашите заявки (по една на ред):")