Update app.py

app.py

@@ -9,7 +9,7 @@ submit = st.button("Submit")
 
 tokenizer = AutoTokenizer.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
 model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
-model = PeftModel.from_pretrained(model, "shionhonda/tiny-llama-reviewer2-1.1B-dpo-lora")
+model = PeftModel.from_pretrained(model, "shionhonda/tiny-llama-reviewer2-1.1B-kto-lora")
 
 prompt = tokenizer.apply_chat_template([
         {"role": "system", "content": "You are an experienced researcher and a reviewer of scientific papers. Given a title of the paper, write a review about it in one sentence."},