Update app.py

app.py

@@ -1,5 +1,11 @@
 import subprocess
 subprocess.run("pip install gradio==4.44.0", shell=True)
+subprocess.run(
+    "pip install flash-attn --no-build-isolation",
+    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    shell=True,
+)
+
 
 from PIL import Image
 import gradio as gr
@@ -21,12 +27,6 @@ from emu3.mllm.processing_emu3 import Emu3Processor
 import io
 import base64
 
-subprocess.run(
-    "pip3 install flash-attn --no-build-isolation",
-    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
-    shell=True,
-)
-
 def image2str(image):
     buf = io.BytesIO()
     image.save(buf, format="PNG")