Update app.py

app.py

@@ -2,8 +2,6 @@ import gradio as gr
 import torch
 import random
 from PIL import Image
-import modeling_florence2
-modeling_florence2.DaViT._initialize_weights = lambda self, module=None: None
 from transformers import AutoProcessor, AutoModelForCausalLM
 from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 
@@ -12,6 +10,7 @@ REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
 # Florence-2 로드
 device = "cuda" if torch.cuda.is_available() else "cpu"
 florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True, ignore_mismatched_sizes=True, ).to(device).eval()
+florence_model.DaViT._initialize_weights = lambda self, module=None: None
 florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
 
 # Stable Diffusion TurboX 로드