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/pegasus-reddit-7e05")
-model = AutoModelForSeq2SeqLM.from_pretrained("QuickRead/pegasus-reddit-7e05")
+tokenizer = AutoTokenizer.from_pretrained("SophieTr/PPO-policy_v2")
+model = AutoModelForSeq2SeqLM.from_pretrained("SophieTr/PPO-policy_v2")
 
 def preprocess(inp):
     input_ids = tokenizer(inp, return_tensors="pt").input_ids