Spaces:

SophieTr
/

TextSummarizationDemo

Runtime error

SophieTr commited on Mar 15, 2022

Commit

3d4f9fa

1 Parent(s): 485f9a0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 # preprocess input
 # return input_ids matrix
-tokenizer = AutoTokenizer.from_pretrained("QuickRead/pegasus-reddit-6.35e5")
-model = AutoModelForSeq2SeqLM.from_pretrained("QuickRead/pegasus-reddit-6.35e5")
 def preprocess(inp):
     input_ids = tokenizer(inp, return_tensors="pt").input_ids

 # preprocess input
 # return input_ids matrix
+tokenizer = AutoTokenizer.from_pretrained("QuickRead/pegasus-reddit-7e05")
+model = AutoModelForSeq2SeqLM.from_pretrained("QuickRead/pegasus-reddit-7e05")
 def preprocess(inp):
     input_ids = tokenizer(inp, return_tensors="pt").input_ids