Hugging Face's logo

Spaces:
valurank
/
Headline_generator
Runtime error

App Files Community
abdulmatinomotoso commited on
Commit
bdece0b
•
1 Parent(s): e79d3d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ from transformers import (
35
  )
36
 
37
  #initializing the tokenizer and the model
38
- tokenizer = AutoTokenizer.from_pretrained("abdulmatinomotoso/pegasus-samsum")
39
- model = AutoModelForSeq2SeqLM.from_pretrained("abdulmatinomotoso/pegasus-samsum")
40
 
41
  #Defining a function to generate the headlines
42
  def headline_generator_2(file):
 
35
  )
36
 
37
  #initializing the tokenizer and the model
38
+ tokenizer = AutoTokenizer.from_pretrained("valurank/headline_generator_baseline")
39
+ model = AutoModelForSeq2SeqLM.from_pretrained("valurank/headline_generator_baseline")
40
 
41
  #Defining a function to generate the headlines
42
  def headline_generator_2(file):