tolulope commited on
Commit
30ae265
·
verified ·
1 Parent(s): ead7acd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
9
  model_id = "JerniganLab/interviews-and-qa"
10
  base_model = "meta-llama/Meta-Llama-3-8B-Instruct"
11
 
12
- llama_model = transformers.AutoModelForCausalLM(base_model)
13
 
14
  pipeline = transformers.pipeline(
15
  "text-generation",
 
9
  model_id = "JerniganLab/interviews-and-qa"
10
  base_model = "meta-llama/Meta-Llama-3-8B-Instruct"
11
 
12
+ llama_model = transformers.AutoModelForCausalLM.from_pretrained(base_model)
13
 
14
  pipeline = transformers.pipeline(
15
  "text-generation",