Update app.py

app.py

@@ -21,6 +21,19 @@ snapshot_download(
     local_dir = "./inference/xcodec_mini_infer"
 )
 
+# Install required package
+def install_flash_attn():
+    try:
+        print("Installing flash-attn...")
+        subprocess.run(
+            ["pip", "install", "flash-attn", "--no-build-isolation"], 
+            check=True
+        )
+        print("flash-attn installed successfully!")
+    except subprocess.CalledProcessError as e:
+        print(f"Failed to install flash-attn: {e}")
+        exit(1)
+
 # Change to the "inference" directory
 inference_dir = "./inference"
 try: