ariG23498 commited on
Commit
bd41503
1 Parent(s): ee678e0

chore: fixing secrets

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ embedding_model = SentenceTransformer(model_name_or_path="all-mpnet-base-v2", de
12
  tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")
13
  model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it", torch_dtype=torch.bfloat16, device_map="auto")
14
 
15
- @spaces.GPU(duration=500)
16
  def process_query(query):
17
  text_embeddings = embedding_model.encode(dataset["train"]["text"])
18
  query_embedding = embedding_model.encode(query)
 
12
  tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")
13
  model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it", torch_dtype=torch.bfloat16, device_map="auto")
14
 
15
+ @spaces.GPU(duration=300)
16
  def process_query(query):
17
  text_embeddings = embedding_model.encode(dataset["train"]["text"])
18
  query_embedding = embedding_model.encode(query)