Update README.md

README.md

@@ -34,11 +34,8 @@ You may also need to run some `pip install` statements at the beginning of the s
 ```python
 !pip install accelerate -U -q
 ```
-
-A similar model can also be trained using the Github with a training script and conda env YAML, which can be
-[found here](https://github.com/Amelie-Schreiber/esm2_LoRA_binding_sites/tree/main). This version uses wandb sweeps for hyperparameter search. 
-However, it does not use class weighting. 
-
+Try to improve upon these metrics by adjusting the hyperparameters:
+```
 ```
 {'eval_loss': 0.49476009607315063,
 'eval_precision': 0.14372964169381108,
@@ -48,6 +45,11 @@ However, it does not use class weighting.
 'epoch': 15.0}
 ```
 
+A similar model can also be trained using the Github with a training script and conda env YAML, which can be
+[found here](https://github.com/Amelie-Schreiber/esm2_LoRA_binding_sites/tree/main). This version uses wandb sweeps for hyperparameter search. 
+However, it does not use class weighting. 
+
+
 ### Framework versions
 
 - PEFT 0.4.0