Update app.py

app.py

@@ -1,8 +1,11 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import LlamaForCausalLM, LlamaTokenizer
 
 model_name = "mlabonne/llama-2-7b-guanaco"
 
-try:
-    model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=False, device='cpu', memory_limit=1024*16)
-except Exception as e:
-    print(f"Error loading the model: {e}")
+# Load the model
+model = LlamaForCausalLM.from_pretrained(model_name)
+
+# Load the tokenizer
+tokenizer = LlamaTokenizer.from_pretrained(model_name)
+
+# Now, you can use the model and tokenizer as needed