Vaishnavi-15 commited on
Commit
d378450
·
verified ·
1 Parent(s): cb552e6

Update app.py

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