Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import subprocess
 import torch
 from PIL import Image
-from transformers import AutoModel, AutoProcessor, AutoModelForCausalLM
+from transformers import AutoProcessor, AutoModelForCausalLM
 
 # import os
 # import random
@@ -13,7 +13,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
 
 # Initialize Florence model
 device = "cuda" if torch.cuda.is_available() else "cpu"
-florence_model = AutoModel.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to(device).eval()
+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)
 
 # api_key = os.getenv("HF_READ_TOKEN")