Update app.py

app.py

@@ -6,6 +6,7 @@ import re
 from io import BytesIO
 import types
 import sys
+import importlib.machinery
 
 # Force CPU-only & disable bitsandbytes CUDA checks in this environment
 os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")
@@ -18,6 +19,7 @@ from PIL import Image
 from torchvision.transforms.functional import InterpolationMode
 import gradio as gr
 
+# Stub bitsandbytes and flash_attn to avoid GPU driver checks in CPU-only environments
 # Stub bitsandbytes and flash_attn to avoid GPU driver checks in CPU-only environments
 fake_bnb = types.ModuleType("bitsandbytes")
 def _bnb_unavailable(*args, **kwargs):
@@ -27,6 +29,8 @@ fake_bnb._bnb_unavailable = _bnb_unavailable
 sys.modules["bitsandbytes"] = fake_bnb
 
 fake_flash = types.ModuleType("flash_attn")
+# set a valid __spec__ so importlib.util.find_spec('flash_attn') does not fail
+fake_flash.__spec__ = importlib.machinery.ModuleSpec("flash_attn", loader=None)
 sys.modules["flash_attn"] = fake_flash
 
 from transformers import AutoModel, AutoTokenizer