Hugging Face's logo

Spaces:
SophieTr
/
TextSummarizationDemo
Runtime error

App Files Community
SophieTr commited on
Commit
6842f32
·
1 Parent(s): 06257b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  # preprocess input
5
  # return input_ids matrix
6
- tokenizer = AutoTokenizer.from_pretrained("QuickRead/PPO-policy_v3")
7
  model = AutoModelForSeq2SeqLM.from_pretrained("QuickRead/PPO-policy_v3")
8
 
9
  def preprocess(inp):
 
3
 
4
  # preprocess input
5
  # return input_ids matrix
6
+ tokenizer = AutoTokenizer.from_pretrained("QuickRead/PPO_training")
7
  model = AutoModelForSeq2SeqLM.from_pretrained("QuickRead/PPO-policy_v3")
8
 
9
  def preprocess(inp):