LovnishVerma's picture
Update app.py
5130d64
raw
history blame
No virus
238 Bytes
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "mlabonne/llama-2-7b-guanaco"
try:
model = AutoModelForCausalLM.from_pretrained(model_name)
except Exception as e:
print(f"Error loading the model: {e}")