Spaces:
Runtime error
Runtime error
abdulmatinomotoso
commited on
Commit
•
4c4e6d4
1
Parent(s):
174033e
Update app.py
Browse files
app.py
CHANGED
@@ -36,9 +36,9 @@ from transformers import (
|
|
36 |
)
|
37 |
|
38 |
#initializing the tokenizer and the model
|
39 |
-
model_type_2 ="
|
40 |
-
tokenizer_2 = AutoTokenizer.from_pretrained(model_type_2)
|
41 |
-
model_2 = AutoModelForSeq2SeqLM.from_pretrained(model_type_2)
|
42 |
|
43 |
#Defining a function to generate the headlines
|
44 |
def headline_generator_2(file):
|
|
|
36 |
)
|
37 |
|
38 |
#initializing the tokenizer and the model
|
39 |
+
model_type_2 ="valurank/pegasus-multi_news-headline"
|
40 |
+
tokenizer_2 = AutoTokenizer.from_pretrained(model_type_2, use_auth_token='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')
|
41 |
+
model_2 = AutoModelForSeq2SeqLM.from_pretrained(model_type_2, use_auth_token='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')
|
42 |
|
43 |
#Defining a function to generate the headlines
|
44 |
def headline_generator_2(file):
|