Update app.py

app.py

@@ -10,13 +10,15 @@ def install_dependencies():
     try:
         # Install torch first
         subprocess.check_call([sys.executable, "-m", "pip", "install", "torch==2.0.1"])
-        # Install other dependencies from requirements.txt
+        # Install flash-attn after torch
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "flash-attn==2.7.2.post1"])
+        # Install other dependencies
         subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
     except subprocess.CalledProcessError as e:
-        print(f"Error occurred during dependency installation: {e}")
+        print(f"Error occurred while installing dependencies: {e}")
         sys.exit(1)
 
-# Run the dependency installer
+# Call the function to install dependencies
 install_dependencies()