Update app.py

app.py

@@ -12,6 +12,15 @@ import subprocess
 
 #os.system("pip install ./flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl")
 subprocess.check_call([sys.executable, "-m", "pip", "install", "./flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl"])
+def ensure_flash_attn():
+    try:
+        import flash_attn
+        print("当前 flash-attn 已安装，版本：", flash_attn.__version__)
+    except ImportError:
+        print("未安装 flash-attn，开始安装...")
+        
+
+ensure_flash_attn()
 
 model_path = snapshot_download(repo_id="Skywork/UniPic2-Metaquery-9B")
 qwen_vl_path = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct-AWQ")