add normalization

app.py

@@ -60,6 +60,13 @@ def siglip_detector(image, texts):
   return probs
 
 
+def normalize_tensor(tensor):
+    # no other normalization works well for visual purposes
+    sum_tensor = torch.sum(tensor)
+    normalized_tensor = tensor / sum_tensor
+
+    return normalized_tensor
+
 def infer_siglip(image, candidate_labels):
   candidate_labels = [label.lstrip(" ") for label in candidate_labels.split(",")]
   siglip_out = siglip_detector(image, candidate_labels)