Soooma commited on
Commit
6e2fb41
·
1 Parent(s): 82de340

GUI change

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)