Update app.py

app.py

@@ -1,5 +1,11 @@
 import sys, types
 
+sys.modules['flash_attn'] = types.ModuleType('flash_attn')
+
+import transformers.dynamic_module_utils as dmu
+dmu.check_imports = lambda *args, **kwargs: []
+
+
 import gradio as gr
 import torch
 import random
@@ -7,8 +13,6 @@ from PIL import Image
 from transformers import AutoProcessor, AutoModelForCausalLM
 from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 
-sys.modules['flash_attn'] = types.ModuleType('flash_attn')
-
 REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
 
 # Florence-2 로드