Update app.py

app.py

@@ -20,6 +20,7 @@ import fitz  # PyMuPDF
 import gradio as gr
 import requests
 import torch
+import subprocess
 from huggingface_hub import snapshot_download
 from PIL import Image, ImageDraw, ImageFont
 from transformers import AutoModelForCausalLM, AutoProcessor, VisionEncoderDecoderModel
@@ -32,6 +33,15 @@ import albumentations as alb
 from albumentations.pytorch import ToTensorV2
 from timm.data.constants import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 
+
+# Attempt to install flash-attn
+try:
+    subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, check=True, shell=True)
+except subprocess.CalledProcessError as e:
+    print(f"Error installing flash-attn: {e}")
+    print("Continuing without flash-attn.")
+
+
 # --- Constants & Global State ---
 MIN_PIXELS = 3136
 MAX_PIXELS = 11289600