Update README.md

README.md

@@ -25,16 +25,12 @@ This model is a fine-tuned version of [microsoft/phi-1_5](https://huggingface.co
 It achieves the following results on the evaluation set:
 - Loss: 2.1587
 
-## Model description
-
-## Sample Code
-
 ### Requirements
 ```python
 !pip install accelerate transformers einops datasets peft bitsandbytes
 ```
 
-### Test Dataset
+## Test Dataset
 If you prefer, you can use test dataset from [zelalt/scientific-papers](https://huggingface.co/datasets/zelalt/scientific-papers)
 or [zelalt/arxiv-papers](https://huggingface.co/datasets/zelalt/arxiv-papers) or read your pdf as text with PyPDF2.PdfReader then give this text to LLM with adding "What is the title of this paper?" prompt.
 
@@ -44,14 +40,14 @@ from datasets import load_dataset
 test_dataset = load_dataset("zelalt/scientific-papers", split='train')
 test_dataset = test_dataset.rename_column('full_text', 'text')
 
-def formatting_prompts_func(example):
+def formatting(example):
     text = f"What is the title of this paper? {example['text'][:180]}\n\nAnswer: "
     return {'text': text}
 
-formatted_dataset = test_dataset.map(formatting_prompts_func)
+formatted_dataset = test_dataset.map(formatting)
 ```
 
-### Inference
+### Sample Code
 ```python
 
 import torch
@@ -79,17 +75,18 @@ text = tokenizer.batch_decode(outputs)[0]
 print(text)
 ```
 
-After running it for the first time and loading the model and tokenizer, you can only run generating part to avoid RAM crash.
+**Notes**
+- After running it for the first time and loading the model and tokenizer, you can only run generating part to avoid RAM crash.
 
 ### Output
 Input:
-```
+```markdown
 What is the title of this paper? Bursting Dynamics of the 3D Euler Equations\nin Cylindrical Domains\nFrançois Golse ∗ †\nEcole Polytechnique, CMLS\n91128 Palaiseau Cedex, France\nAlex Mahalov ‡and Basil Nicolaenko §\n\nAnswer:
 ```
 
 ## Output from LLM:
 
-```
+```markdown
 What is the title of this paper? Bursting Dynamics of the 3D Euler Equations
 in Cylindrical Domains
 François Golse ∗ †