Update app.py

app.py

@@ -8,11 +8,11 @@ from transformers import pipeline,AutoModelForCausalLM, AutoTokenizer
 import torch
 
 
-model_name = "meta-llama/Llama-2-7b-chat-hf"
-access_token = base64.b64decode('aGZfekJxa0pXZEh1bm90UEJXek1mdkdOc096WXdIVVZvYkRwcg==')  # Replace with your token
+model_name = "bmas10/OG_llama2"
+#access_token = base64.b64decode('aGZfekJxa0pXZEh1bm90UEJXek1mdkdOc096WXdIVVZvYkRwcg==')  # Replace with your token
 
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_token)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", use_auth_token=access_token)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")#, use_auth_token=access_token)
 # Load model and tokenizer