Update app.py

app.py

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