ASledziewska commited on
Commit
a6a602a
1 Parent(s): e6c8f43

updated API key reference

Browse files
Files changed (1) hide show
  1. llama_guard.py +1 -3
llama_guard.py CHANGED
@@ -2,7 +2,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import torch
3
 
4
  # Docs:- https://huggingface.co/meta-llama/LlamaGuard-7b
5
- from dotenv import load_dotenv, find_dotenv
6
  import os
7
  import requests
8
 
@@ -10,8 +9,7 @@ model_id = "meta-llama/LlamaGuard-7b"
10
 
11
 
12
  # Please ensure you have a .env file available with 'HUGGINGFACEHUB_API_TOKEN'
13
- load_dotenv(find_dotenv())
14
- HUGGINGFACEHUB_API_TOKEN = os.environ["HUGGINGFACEHUB_API_TOKEN"]
15
 
16
  # updated on March 24th
17
 
 
2
  import torch
3
 
4
  # Docs:- https://huggingface.co/meta-llama/LlamaGuard-7b
 
5
  import os
6
  import requests
7
 
 
9
 
10
 
11
  # Please ensure you have a .env file available with 'HUGGINGFACEHUB_API_TOKEN'
12
+ HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACE_LLAMAGAURD_API_TOKEN")
 
13
 
14
  # updated on March 24th
15