Update app.py

app.py

@@ -5,11 +5,12 @@ from PIL import Image
 from transformers import AutoProcessor, AutoModelForCausalLM
 from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 
+REVISION = "9d515616ecb4f49eecc160bd60eec62356e5f858"
 
 # Florence-2 로드
 device = "cuda" if torch.cuda.is_available() else "cpu"
-florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to(device).eval()
-florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True)
+florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True).to(device).eval()
+florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
 
 # Stable Diffusion TurboX 로드
 model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"