Update app.py

app.py

@@ -1,4 +1,6 @@
 import subprocess
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
 import os
 
 import gradio as gr
@@ -11,7 +13,6 @@ from pix2struct.modeling import Pix2StructModel
 from pix2struct.processing import extract_patches
 from pix2struct.inference import ask_generator, generate, DocumentQueries, DocumentQuery
 
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 hub_token = os.environ.get('HUB_TOKEN')
 model_path = snapshot_download('artyomxyz/pix2struct-docmatix', use_auth_token=hub_token)