tolulope commited on
Commit
6535a3b
·
verified ·
1 Parent(s): 95e82c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,10 +7,11 @@ import os
7
  HF_TOKEN = os.environ.get("HF_TOKEN")
8
 
9
  model_id = "JerniganLab/interviews-and-qa"
 
10
 
11
  pipeline = transformers.pipeline(
12
  "text-generation",
13
- model="meta-llama/Meta-Llama-3-8B-Instruct",
14
  model_kwargs={"torch_dtype": torch.bfloat16},
15
  device="cuda",
16
  )
 
7
  HF_TOKEN = os.environ.get("HF_TOKEN")
8
 
9
  model_id = "JerniganLab/interviews-and-qa"
10
+ base_model = "meta-llama/Meta-Llama-3-8B-Instruct"
11
 
12
  pipeline = transformers.pipeline(
13
  "text-generation",
14
+ model=base_model,
15
  model_kwargs={"torch_dtype": torch.bfloat16},
16
  device="cuda",
17
  )