Update app.py

app.py

@@ -4,7 +4,9 @@ import spaces
 import re
 from PIL import Image
 
-# No need to install flash-attn since it's GPU-specific
+# Install the necessary packages
+import subprocess
+subprocess.run('pip install flash-attn einops --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 model = AutoModelForCausalLM.from_pretrained('gokaygokay/Florence-2-SD3-Captioner', trust_remote_code=True).eval()