Update app.py

app.py

@@ -1,3 +1,20 @@
+import subprocess
+
+def uninstall_package(package_name):
+    try:
+        subprocess.check_call([f"pip uninstall {package_name} -y"], shell=True)
+        print(f"Successfully uninstalled {package_name}.")
+    except subprocess.CalledProcessError as e:
+        print(f"Failed to uninstall {package_name}. Error: {e}")
+
+# Specify the package name
+package_name = "flash-attn"
+
+# Call the function to uninstall the package
+uninstall_package(package_name)
+
+
+
 import gradio as gr
 import requests
 import os