put installation and imports at beginning of app

app.py

@@ -1,14 +1,22 @@
-# non-GPU imports
+# install custom package
+import os
+os.system("pip install --no-deps ./mosaicfm-0.1.2-py3-none-any.whl")
+
+# imports
 import gc
-import spaces
 import tempfile
+import torch
 import gradio as gr
 import anndata as ad
 import pandas as pd
 import numpy as np
 import scanpy as sc
 from pathlib import Path
+from composer import Trainer
 from omegaconf import OmegaConf as om
+from mosaicfm.model.model import ComposerSCGPTModel
+from mosaicfm.data import CountDataset, DataCollator
+from mosaicfm.tokenizer import GeneVocab
 
 # hardcoded configuration
 EMB_KEY = "X_tx1-70m"
@@ -56,21 +64,8 @@ def _save_outputs(adata, emb):
     return str(parquet_path), str(out_h5ad)
 
 # compute embeddings
-@spaces.GPU
 def _embed(adata_bytes, layer_name, feature_col, batch_size):
 
-    # install package
-    import os
-    # os.system("pip install --no-build-isolation flash-attn")
-    os.system("pip install --no-deps ./mosaicfm-0.1.2-py3-none-any.whl")
-
-    # imports inside GPU scope
-    import torch
-    from composer import Trainer
-    from mosaicfm.model.model import ComposerSCGPTModel
-    from mosaicfm.data import CountDataset, DataCollator
-    from mosaicfm.tokenizer import GeneVocab
-
     # retrieve AnnData from bytes
     with tempfile.TemporaryDirectory() as td:
         fpath = Path(td) / "input.h5ad"