orionweller commited on
Commit
00b89f1
·
1 Parent(s): 9cbd6cd

don't cache examples

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -132,7 +132,7 @@ def encode_queries(dataset_name, postfix):
132
  input_texts = [f"query: {query.strip()} {postfix}".strip() for query in queries[dataset_name]]
133
 
134
  encoded_embeds = []
135
- batch_size = 2
136
 
137
  for start_idx in tqdm.tqdm(range(0, len(input_texts), batch_size), desc="Encoding queries"):
138
  batch_input_texts = input_texts[start_idx: start_idx + batch_size]
@@ -209,7 +209,7 @@ iface = gr.Interface(
209
  ["scifact", ""],
210
  ["scifact", "Think carefully about these conditions when determining relevance"]
211
  ],
212
- cache_examples=True,
213
  )
214
 
215
  # Launch the interface
 
132
  input_texts = [f"query: {query.strip()} {postfix}".strip() for query in queries[dataset_name]]
133
 
134
  encoded_embeds = []
135
+ batch_size = 32
136
 
137
  for start_idx in tqdm.tqdm(range(0, len(input_texts), batch_size), desc="Encoding queries"):
138
  batch_input_texts = input_texts[start_idx: start_idx + batch_size]
 
209
  ["scifact", ""],
210
  ["scifact", "Think carefully about these conditions when determining relevance"]
211
  ],
212
+ cache_examples=False,
213
  )
214
 
215
  # Launch the interface