app.py

import transformers 
from transformers import AutoModelForCausalLM
model_id = "llama/Llama-2-7b-chat-hf"

model = AutoModelForCausalLM.from_pretrained(model_id)

text = "Hello"

outputs = model.generate(text)
print(outputs)