Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
tkoooo authored May 1, 2024
1 parent 47db6d5 commit 4975d56
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -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!

Expand Down

0 comments on commit 4975d56

Please sign in to comment.