chore: fixing secrets
Browse files
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=
|
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)
|