Spaces:

SophieTr
/

TextSummarizationDemo

Runtime error

SophieTr commited on Apr 22, 2022

Commit

14b09b6

1 Parent(s): 18bd811

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 # preprocess input
 # return input_ids matrix
 tokenizer = AutoTokenizer.from_pretrained("QuickRead/PPO_training")
-model = AutoModelForSeq2SeqLM.from_pretrained("SophieTr/PPO-policy_v2")
 def preprocess(inp):
     input_ids = tokenizer(inp, return_tensors="pt").input_ids

 # preprocess input
 # return input_ids matrix
 tokenizer = AutoTokenizer.from_pretrained("QuickRead/PPO_training")
+model = AutoModelForSeq2SeqLM.from_pretrained("SophieTr/PPO_training")
 def preprocess(inp):
     input_ids = tokenizer(inp, return_tensors="pt").input_ids