Vaishnavi-15 commited on
Commit
9d2f6bd
·
verified ·
1 Parent(s): d3d22f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,10 +4,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
4
 
5
  #login(token="hf_VExbFezQQyzOnbpBoRgNxXjiRfMFTGUyj")
6
 
 
 
7
  # Load model and tokenizer
8
  model_name = "meta-llama/CodeLlama-7b-hf"
9
- tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
10
- model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
11
 
12
  # Define the inference function
13
  def generate_code(prompt):
 
4
 
5
  #login(token="hf_VExbFezQQyzOnbpBoRgNxXjiRfMFTGUyj")
6
 
7
+ my_token="hf_VExbFezQQyzOnbpBoRgNxXjiRfMFTGUyj"
8
+
9
  # Load model and tokenizer
10
  model_name = "meta-llama/CodeLlama-7b-hf"
11
+ tokenizer = AutoTokenizer.from_pretrained(model_name, token=my_token)
12
+ model = AutoModelForCausalLM.from_pretrained(model_name, token=my_token)
13
 
14
  # Define the inference function
15
  def generate_code(prompt):