Update app.py

app.py

@@ -1,3 +1,5 @@
+import sys, types
+
 import gradio as gr
 import torch
 import random
@@ -5,6 +7,8 @@ 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 로드