update readme

README.md

@@ -3,7 +3,7 @@ title: OmniGen2
 emoji: 👀
 colorFrom: red
 colorTo: yellow
-sdk: docker
+sdk: gradio
 pinned: false
 license: apache-2.0
 short_description: 'OmniGen2: Unified Image Understanding and Generation.'

app.py

@@ -2,6 +2,9 @@ import dotenv
 
 dotenv.load_dotenv(override=True)
 
+import subprocess
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
 import gradio as gr
 
 import os