Update app.py

app.py

@@ -10,6 +10,9 @@ from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse
 import uvicorn
 
+# Set environment variable to handle remote code trust
+os.environ["TRANSFORMERS_OFFLINE"] = "1"
+
 # Configure PyTorch settings
 torch.backends.cuda.enable_flash_sdp(False)
 torch.backends.cuda.enable_math_sdp(True)
@@ -40,14 +43,18 @@ def load_florence_model():
             log_message("[*] Loading Florence-2 model and processor...")
             
             # Load model and processor
+            # Load processor with explicit trust and local files
             processor = AutoProcessor.from_pretrained(
                 MODEL_ID,
-                trust_remote_code=True
+                trust_remote_code=True,
+                local_files_only=True
             )
             
+            # Load model with explicit trust and local files
             model = AutoModelForVision2Seq.from_pretrained(
                 MODEL_ID,
                 trust_remote_code=True,
+                local_files_only=True,
                 torch_dtype=torch.float32
             ).to(DEVICE)