Umair Khan commited on
Commit
9be3c4b
·
1 Parent(s): d4b8585

remove batch_size references

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -22,7 +22,6 @@ from mosaicfm.tokenizer import GeneVocab
22
 
23
  # hardcoded configuration
24
  EMB_KEY = "X_tx1-70m"
25
- BATCH_SIZE = 64
26
  APP_TITLE = "Tx1-70M Embeddings"
27
  APP_DESC = """
28
  Upload an AnnData, compute Tx1-70M embeddings,
@@ -267,7 +266,7 @@ def _embed(adata_bytes, layer_name, feature_col, progress):
267
  return cell_array, layers, var_cols, obs_cols, adata_persisted
268
 
269
  # processing pipeline given user inputs
270
- def run_pipeline(fileobj, layer_choice, var_choice, obs_choice, batch_size=64, progress=gr.Progress(track_tqdm=False)):
271
 
272
  # make sure user inputs exist
273
  if fileobj is None:
@@ -286,7 +285,7 @@ def run_pipeline(fileobj, layer_choice, var_choice, obs_choice, batch_size=64, p
286
  adata_bytes=adata_bytes,
287
  layer_name=(None if layer_choice in [None, "", "<use .X>"] else layer_choice),
288
  feature_col=(None if var_choice in [None, ""] else var_choice),
289
- batch_size=int(batch_size),
290
  )
291
 
292
  # rebuild AnnData on CPU
 
22
 
23
  # hardcoded configuration
24
  EMB_KEY = "X_tx1-70m"
 
25
  APP_TITLE = "Tx1-70M Embeddings"
26
  APP_DESC = """
27
  Upload an AnnData, compute Tx1-70M embeddings,
 
266
  return cell_array, layers, var_cols, obs_cols, adata_persisted
267
 
268
  # processing pipeline given user inputs
269
+ def run_pipeline(fileobj, layer_choice, var_choice, obs_choice, progress=gr.Progress(track_tqdm=False)):
270
 
271
  # make sure user inputs exist
272
  if fileobj is None:
 
285
  adata_bytes=adata_bytes,
286
  layer_name=(None if layer_choice in [None, "", "<use .X>"] else layer_choice),
287
  feature_col=(None if var_choice in [None, ""] else var_choice),
288
+ progress=progress
289
  )
290
 
291
  # rebuild AnnData on CPU