Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-tokenizer = AutoTokenizer.from_pretrained("QuickRead/PPO-policy_v3")
+tokenizer = AutoTokenizer.from_pretrained("SophieTr/PP0_training")
 
-model = AutoModelForSeq2SeqLM.from_pretrained("QuickRead/PPO-policy_v3")
+model = AutoModelForSeq2SeqLM.from_pretrained("SophieTr/PP0_rm_v1")
 
 def preprocess(inp):
     input_ids = tokenizer(inp, return_tensors="pt").input_ids