Upload folder using huggingface_hub

api/startup.py

@@ -36,6 +36,7 @@ def download_artifacts():
 
     files_to_download = [
         ("models/pca_256.pkl",              "data/pca_256.pkl"),
+        ("models/midas_small.onnx",         "data/midas_small.onnx"),
         ("configs/thresholds.json",          "data/thresholds.json"),
         ("graph/knowledge_graph.json",       "data/knowledge_graph.json"),
         ("indexes/index1_category.faiss",    "data/index1_category.faiss"),

src/patchcore.py

@@ -149,7 +149,7 @@ class PatchCoreExtractor:
 
         # Gaussian smoothing
         if smooth:
-            heatmap = scipy.ndimage.gaussian_filter(heatmap, sigma=4)
+            heatmap = scipy.ndimage.gaussian_filter(heatmap, sigma=2)
 
         # Normalise to [0, 1]
         h_min, h_max = heatmap.min(), heatmap.max()