Update app.py

app.py

@@ -10,8 +10,23 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer
 import os
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-subprocess.run('pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git+https://github.com/NVIDIA/apex.git
-', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
+
+def install_apex():
+    cmd = [
+        "pip", "install", "-v", "--disable-pip-version-check", "--no-cache-dir",
+        "--no-build-isolation", "--config-settings", "--build-option=--cpp_ext",
+        "--config-settings", "--build-option=--cuda_ext",
+        "git+https://github.com/NVIDIA/apex.git"
+    ]
+    subprocess.run(cmd, check=True)
+
+try:
+    import apex
+except ImportError:
+    print("apex not installed, installing now...")
+    install_apex()
+    import apex  # Try to import again after installation
 
 
 def download_t5_model(model_id, save_directory):