Update app.py

app.py

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