Update app.py

app.py

@@ -9,20 +9,25 @@ import spaces
 from transformers import T5ForConditionalGeneration, T5Tokenizer
 import os
 
+def install_apex():
+    # Install Apex in editable mode from the specified GitHub repository
+    cmd = [
+        'pip', 'install', '-v','--no-cache-dir', '--no-build-isolation',
+        '--config-settings', 'build-option=--cpp_ext', '--config-settings',
+        'build-option=--cuda_ext', 'git+https://github.com/sandeshrajbhandari/apex.git#egg=apex'
+    ]
+    subprocess.run(cmd, check=True, capture_output=True)
+
+try:
+    import apex
+except ModuleNotFoundError:
+    print("Apex not found, installing...")
+    install_apex()
+    # Try to import Apex again after installation
+    import apex
+
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-install_options = [
-    "--disable-pip-version-check",
-    "--no-cache-dir",
-    "--no-build-isolation",
-    "--config-settings", "--build-option=--cpp_ext",
-    "--config-settings", "--build-option=--cuda_ext"
-]
-print("tseting")
-subprocess.run(
-    ["pip", "install", "-v"] + install_options + ["git+https://github.com/sandeshrajbhandari/apex.git#egg=apex"],
-    check=True,
-    capture_output=True
-)
+
 def download_t5_model(model_id, save_directory):
     # Modelin tokenizer'ını ve modeli indir
     model = T5ForConditionalGeneration.from_pretrained(model_id)