Hugging Face's logo

Spaces:
Soooma
/
titles_gen
Runtime error

App Files Community
Soooma commited on
Commit
0b02152
·
1 Parent(s): b743bc7

Minor change to test my titles_gen saved model!

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -4,8 +4,11 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
  import nltk
5
  nltk.download('punkt')
6
 
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
 
 
4
  import nltk
5
  nltk.download('punkt')
6
 
7
+ #tokenizer = AutoTokenizer.from_pretrained("fabiochiu/t5-small-medium-title-generation")
8
+ #model = AutoModelForSeq2SeqLM.from_pretrained("fabiochiu/t5-small-medium-title-generation")
9
+
10
+ tokenizer = AutoTokenizer.from_pretrained("Soooma/titles_gen")
11
+ model = AutoModelForSeq2SeqLM.from_pretrained("Soooma/titles_gen")
12
 
13
  text = st.text_area('Enter an abstract to summerize, i.e. generate a title!', height=440)
14