Refactor and add optional wandb logging

[morganmcg1]

Refactors:

• The model calls to use tenacity for retries
• Removes duplication of code
• Pulls out config variables into their own cell

Adds:

• optional logging of config, prompts and results to Weights & Biases to better keep track of your work and results