AgaMiko commited on
Commit
2697aa2
·
1 Parent(s): e85d00a

update model, delete sidebar

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -2,8 +2,8 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer
2
  import streamlit as st
3
  from PIL import Image
4
 
5
- tokenizer= T5Tokenizer.from_pretrained("Voicelab/vlt5-base-keywords")
6
- model = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords")
7
 
8
  img_full = Image.open("images/vl-logo-nlp-blue.png")
9
  img_short = Image.open("images/sVL-NLP-short.png")
@@ -31,7 +31,6 @@ def trim_length():
31
 
32
 
33
  if __name__ == "__main__":
34
- st.sidebar.image(img_short)
35
  st.image(img_full)
36
  st.title('VLT5 - keywords generation')
37
 
 
2
  import streamlit as st
3
  from PIL import Image
4
 
5
+ tokenizer= T5Tokenizer.from_pretrained("Voicelab/vlt5-base-keywords-v4_3-en")
6
+ model = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords-v4_3-en")
7
 
8
  img_full = Image.open("images/vl-logo-nlp-blue.png")
9
  img_short = Image.open("images/sVL-NLP-short.png")
 
31
 
32
 
33
  if __name__ == "__main__":
 
34
  st.image(img_full)
35
  st.title('VLT5 - keywords generation')
36