Update app.py

app.py

@@ -12,7 +12,7 @@ import gradio as gr
 import torch
 import random
 from PIL import Image
-from transformers import AutoProcessor, AutoModelForCausalLM
+from transformers import AutoProcessor, AutoModelForSeq2SeqLM
 from diffusers import DiffusionPipeline
 try:
     from diffusers import FlowMatchEulerDiscreteScheduler
@@ -46,7 +46,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, torch_dtype=torch.float16)
+florence_model = AutoModelForSeq2SeqLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True, torch_dtype=torch.float16)
 florence_model.to("cpu")
 florence_model.eval()
 florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)