piazzola/llama-2-7b-address

This model is a fine-tuned version of NousResearch/Llama-2-7b-hf on the addressWithContext dataset.

from transformers import pipeline
nlp = pipeline("text-generation", 
                model="piazzola/llama-2-7b-address",
                max_new_tokens = 30)

context = "Landlord hereby demises unto Tenant, and Tenant hereby leases from Landlord for the terms and upon the conditions set forth in this Lease 30,600 square feet of space in the building located at 15 Firstfield Road, Gaithersburg, Maryland, the Building, as set forth on Exhibit A, hereto attached, said space being referred to as the Premises."
LLAMA_PROMPT_TEST = (
'[INST] <<SYS>>\n'
'You are a helpful and respectful assistant '
'for getting address information from a text.\n'
'<</SYS>>\n'
'\n'
'Given the following context, answer the following question.\n'
'Context: {Context}\n'
'\n'
'Question:\n'
'What is the address in the given context?\n'
'\n'
'[/INST]\n'
)
LLAMA_PROMPT_TEST = LLAMA_PROMPT_TEST.format(Context=context)

nlp(LLAMA_PROMPT_TEST)