Update app.py

app.py

@@ -1,10 +1,19 @@
+import os
+
+# ── Force fully offline mode BEFORE any transformers import ───────
+# The deployment environment has no internet access. These env vars
+# prevent transformers / huggingface_hub from attempting any network
+# calls (model card fetches, config resolution, token checks, etc.)
+os.environ["TRANSFORMERS_OFFLINE"] = "1"
+os.environ["HF_DATASETS_OFFLINE"]  = "1"
+os.environ["HF_HUB_OFFLINE"]       = "1"
+
 import gradio as gr
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import math
 import numpy as np
-import os
 from PIL import Image, ImageFilter
 from torchvision import transforms, models
 from transformers import AutoModelForImageClassification, AutoImageProcessor
@@ -229,8 +238,11 @@ class TriStreamDetector(nn.Module):
 
 def load_model():
     print("Loading SigLIP backbone...")
+    siglip_path = os.path.join(MODEL_PATH, "siglip_component")
     siglip_base = AutoModelForImageClassification.from_pretrained(
-        os.path.join(MODEL_PATH, "siglip_component")
+        siglip_path,
+        local_files_only=True,
+        ignore_mismatched_sizes=True,
     )
 
     print("Building TriStreamDetector...")