Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import subprocess
 import torch
 from PIL import Image
-from transformers import AutoProcessor, AutoModelForSeq2SeqLM
+from transformers import AutoProcessor, AutoModelForConditionalGeneration
 
 subprocess.run(
     'pip install flash-attn --no-build-isolation',
@@ -12,7 +12,7 @@ subprocess.run(
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = 'microsoft/Florence-2-base-ft'
-florence_model = AutoModelForSeq2SeqLM.from_pretrained(
+florence_model = AutoModelForConditionalGeneration.from_pretrained(
     model,
     trust_remote_code=True).to(device).eval()
 florence_processor = AutoProcessor.from_pretrained(model, trust_remote_code=True)