Spaces:
Runtime error
Runtime error
update model, delete sidebar
Browse files
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 |
|