Update app.py

app.py

@@ -5,13 +5,13 @@ import numpy as np
 from PIL import Image
 import gradio as gr
 import torch
-from transformers import AutoProcessor, AutoModelForCausalLM
+from transformers import AutoProcessor, Florence2ForConditionalGeneration
 
 
 # Load model and processor, enabling trust_remote_code if needed
 model_name = "PJMixers-Images/Florence-2-base-Castollux-v0.5-JPEG-Quality-Detection-v0.1"
-model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True).eval()
-processor = AutoProcessor.from_pretrained(model_name, trust_remote_code=True)
+model = Florence2ForConditionalGeneration.from_pretrained(model_name).eval()
+processor = AutoProcessor.from_pretrained(model_name)
 
 # Set device (GPU if available)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")