Update app.py

app.py

@@ -9,7 +9,18 @@ import spaces
 from transformers import T5ForConditionalGeneration, T5Tokenizer
 import os
 
-print ("starting the app.")
+def check_and_install(package_name):
+    if importlib.util.find_spec(package_name) is None:
+        print(f"{package_name} not installed, installing...")
+        subprocess.run(
+            f'pip install {package_name} --no-build-isolation',
+            env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"},
+            shell=True
+        )
+    else:
+        print(f"{package_name} is already installed.")
+        
+check_and_install('flash_attn')
 
 def download_t5_model(model_id, save_directory):
     # Modelin tokenizer'ını ve modeli indir