Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 import torch.nn as nn
+import torch.serialization  # Added for add_safe_globals
 from torchvision import models, transforms
 from huggingface_hub import hf_hub_download
 from PIL import Image
@@ -24,6 +25,9 @@ CONFIDENCE_THRESHOLD = 0.8  # 80%
 # Mahalanobis distance threshold for OOD detection
 MAHALANOBIS_THRESHOLD = 100.0  # Calibrate this using a validation set
 
+# Allowlist the NumPy global for torch.load
+torch.serialization.add_safe_globals([np._core.multiarray._reconstruct])
+
 # Download model from Hugging Face
 def download_model():
     model_path = hf_hub_download(repo_id="jays009/Resnet3", filename="model.pth")
@@ -58,7 +62,7 @@ def load_main_model(model_path):
 try:
     class_statistics = torch.load("class_statistics.pth", map_location=torch.device("cpu"))
 except FileNotFoundError:
-    logger.error("class_statistics.pth not found. Please run the statistics computation script first.")
+    logger.error("class_statistics.pth not found. Please ensure the file is in the same directory as app.py.")
     raise
 
 # Path to your model