vojay commited on
Commit
37d5637
·
verified ·
1 Parent(s): c5c65c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -17,6 +17,8 @@ adapter_model_id = "vojay/Llama-2-7b-chat-hf-mental-health"
17
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16)
18
  model.load_adapter(adapter_model_id)
19
 
 
 
20
 
21
  def get_base_prompt():
22
  return """
 
17
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16)
18
  model.load_adapter(adapter_model_id)
19
 
20
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
21
+
22
 
23
  def get_base_prompt():
24
  return """