Update app.py

app.py

@@ -1,3 +1,27 @@
+import subprocess
+import sys
+
+# Function to install a package
+def install_package(package_name):
+    subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
+
+# Install the required packages
+try:
+    # Install setuptools
+    install_package("setuptools==74.1.2")
+    
+    # Install wheel
+    install_package("wheel")
+    
+    # Install flash_attn
+    install_package("flash_attn==2.5.8")
+    
+    print("All packages installed successfully.")
+    
+except subprocess.CalledProcessError as e:
+    print(f"Error occurred: {e}")
+
+
 from gradio_client import Client
 import gradio as gr
 import requests