ariG23498 HF staff commited on
Commit
bbd2c0c
·
1 Parent(s): a3f8683

chore: fixing secrets

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ import os
8
 
9
  dataset = load_dataset("ariG23498/pis-blogs-chunked")
10
  embedding_model = SentenceTransformer(model_name_or_path="all-mpnet-base-v2")
11
- tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it", key="HF_TOKEN", value=os.getenv["auth"])
12
- model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it", torch_dtype=torch.bfloat16, key="HF_TOKEN", value=os.getenv["auth"])
13
 
14
  @spaces.GPU
15
  def process_query(query):
 
8
 
9
  dataset = load_dataset("ariG23498/pis-blogs-chunked")
10
  embedding_model = SentenceTransformer(model_name_or_path="all-mpnet-base-v2")
11
+ tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it", key="HF_TOKEN", value=os.getenv("auth"))
12
+ model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it", torch_dtype=torch.bfloat16, key="HF_TOKEN", value=os.getenv("auth"))
13
 
14
  @spaces.GPU
15
  def process_query(query):