piazzola commited on
Commit
6a701ba
·
1 Parent(s): 1f8e02b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -3,7 +3,8 @@ license: cc-by-nc-2.0
3
  ---
4
  This model is a fine-tuned version of [NousResearch/Llama-2-7b-hf](https://huggingface.co/NousResearch/Llama-2-7b-hf) on the [addressWithContext](https://huggingface.co/datasets/piazzola/addressWithContext) dataset.
5
 
6
- ```from transformers import pipeline
 
7
  nlp = pipeline("text-generation",
8
  model="piazzola/llama-2-7b-address",
9
  max_new_tokens = 30)
@@ -25,4 +26,5 @@ LLAMA_PROMPT_TEST = (
25
  )
26
  LLAMA_PROMPT_TEST = LLAMA_PROMPT_TEST.format(Context=context)
27
 
28
- nlp(LLAMA_PROMPT_TEST)```
 
 
3
  ---
4
  This model is a fine-tuned version of [NousResearch/Llama-2-7b-hf](https://huggingface.co/NousResearch/Llama-2-7b-hf) on the [addressWithContext](https://huggingface.co/datasets/piazzola/addressWithContext) dataset.
5
 
6
+ ```
7
+ from transformers import pipeline
8
  nlp = pipeline("text-generation",
9
  model="piazzola/llama-2-7b-address",
10
  max_new_tokens = 30)
 
26
  )
27
  LLAMA_PROMPT_TEST = LLAMA_PROMPT_TEST.format(Context=context)
28
 
29
+ nlp(LLAMA_PROMPT_TEST)
30
+ ```