Spaces:
Runtime error
Runtime error
GUI change
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ nltk.download('punkt')
|
|
7 |
tokenizer = AutoTokenizer.from_pretrained("fabiochiu/t5-small-medium-title-generation")
|
8 |
model = AutoModelForSeq2SeqLM.from_pretrained("fabiochiu/t5-small-medium-title-generation")
|
9 |
|
10 |
-
text = st.text_area('Enter an abstract to summerize, i.e. generate a title!')
|
11 |
|
12 |
max_input_length = 512
|
13 |
|
@@ -18,6 +18,6 @@ if text:
|
|
18 |
decoded_output = tokenizer.batch_decode(output, skip_special_tokens=True)[0]
|
19 |
predicted_title = nltk.sent_tokenize(decoded_output.strip())[0]
|
20 |
|
21 |
-
html_string = f"<h3>{predicted_title}</h3>"
|
22 |
|
23 |
st.markdown(html_string, unsafe_allow_html=True)
|
|
|
7 |
tokenizer = AutoTokenizer.from_pretrained("fabiochiu/t5-small-medium-title-generation")
|
8 |
model = AutoModelForSeq2SeqLM.from_pretrained("fabiochiu/t5-small-medium-title-generation")
|
9 |
|
10 |
+
text = st.text_area('Enter an abstract to summerize, i.e. generate a title!', height=440)
|
11 |
|
12 |
max_input_length = 512
|
13 |
|
|
|
18 |
decoded_output = tokenizer.batch_decode(output, skip_special_tokens=True)[0]
|
19 |
predicted_title = nltk.sent_tokenize(decoded_output.strip())[0]
|
20 |
|
21 |
+
html_string = f"<h3>Predicted title :{predicted_title}</h3>"
|
22 |
|
23 |
st.markdown(html_string, unsafe_allow_html=True)
|