From 4975d564b8139b6ca38c6e714cc9ec1a7e16ee5f Mon Sep 17 00:00:00 2001
From: tkoooo <100322984+tkoooo@users.noreply.github.com>
Date: Tue, 30 Apr 2024 23:07:24 -0500
Subject: [PATCH] Update README.md

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 52c86f8..a3bb7e9 100644
--- a/README.md
+++ b/README.md
@@ -1,8 +1,8 @@
 # CMPS 6730 Project Pickup
 # Mackenzie Bookamer and Thalia Koutsougeras
 
-*Are you a graph? 
-Because you’ve got my heart all connected.*
+**Are you a graph? 
+Because you’ve got my heart all connected.**
 
 That's the power of the *PickupMaster 9000*. This CS/math-themed pickup line generator is here to meet your needs, whether it's to complete the end of your pickup line or to practice honing your skills with you. By fine-tuning the pre-trained language model (PLM) GPT-2 from OpenAI with a wide variety of math and computer science pickup lines, we were able to generate coherent and subject-relevant pickup lines. We pull the PLM from the HuggingFace model hub and use the GPT-2 Tokenizer as well as Transformers Trainer class to process and train our data, then subsequently feed user input back through our fine-tuned model to generate a response. To facilitate user interaction with our model, we created a Flask app that takes in user input and generates 2 pickup lines from our fine tuned model - better than the PLM without fine-tuning!