Update README.md

README.md

@@ -24,8 +24,12 @@ To validate the results the Antimicrobial Peptide Scanner vr.2 (https://www.dvel
 
 ## Training and evaluation data
 
-run_clm.py
+AmpGPT2 was trained using 32014 AMP sequences from the Compass (https://compass.mathematik.uni-marburg.de/) database. 
 
+## How to use AmpGPT2 
+```
+from transformers import pipeline from transformers import GPT2LMHeadModel, GPT2Tokenizer ampgpt2 = pipeline('text-generation', model="wabu/AmpGPT2") model_amp = GPT2LMHeadModel.from_pretrained('wabu/AmpGPT2') tokenizer_amp = GPT2Tokenizer.from_pretrained('wabu/AmpGPT2') amp_sequences = ampgpt2( "", do_sample=True, repetition_penalty=1.2, num_return_sequences=10, eos_token_id=0 ) for i, seq in enumerate(amp_sequences): sequence_identifier = f"Sequence_{i + 1}" sequence = seq['generated_text'].replace('','').strip() print(f">{sequence_identifier}\n{sequence}")
+```
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -38,11 +42,11 @@ The following hyperparameters were used during training:
 - num_epochs: 50.0
 
 ### Training results
-tr
+these are the training losses after the final epoch
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 3.7948        | 50.0  | 7400 | 3.9890          | 0.4213   |
+| Training Loss | Epoch | Validation Loss | Accuracy |
+|:-------------:|:-----:|:---------------:|:--------:|
+| 3.7948        | 50.0  | 3.9890          | 0.4213   |
 
 
 ### Framework versions