Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import sys
-import subprocess
 from threading import Thread
 from typing import Iterable
 from huggingface_hub import snapshot_download
@@ -21,14 +20,6 @@ from gradio.themes.utils import colors, fonts, sizes
 
 # --- Theme and CSS Definition ---
 
-# 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.")
-
-
 colors.steel_blue = colors.Color(
     name="steel_blue",
     c50="#EBF3F8",
@@ -158,14 +149,12 @@ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     torch_dtype=torch.float16
 ).to(device).eval()
 
-import flash_attn_2_cuda as flash_attn_gpu
-
 # Load Dots.OCR from the local, patched directory
 MODEL_PATH_D = model_path_d_local
 processor_d = AutoProcessor.from_pretrained(MODEL_PATH_D, trust_remote_code=True)
 model_d = AutoModelForCausalLM.from_pretrained(
     MODEL_PATH_D,
-    attn_implementation="flash_attention_2",
+    attn_implementation="eager",
     torch_dtype=torch.bfloat16,
     device_map="auto",
     trust_remote_code=True