fix overflow issue

app.py

@@ -15,7 +15,7 @@ from numpy.random import RandomState, SeedSequence
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32") 
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
 
-HYPERDIMS = 4096
+HYPERDIMS = 2048
 VALUE_BITS = 10
 POS_BITS = 9 # CLIP features are 512 dims
 val_bins = np.linspace(start=-1., stop=1., num=2**VALUE_BITS)
@@ -187,7 +187,8 @@ def predict(embeds, exemplars, lut):
     dist_to_ex0 = np.abs(hd_embeds - exemplars[0][np.newaxis, ...]).sum(axis=-1)
     dist_to_ex1 = np.abs(hd_embeds - exemplars[1][np.newaxis, ...]).sum(axis=-1)
     print('dists', dist_to_ex0, dist_to_ex1)
-    preds = np.array([HYPERDIMS - dist_to_ex0, HYPERDIMS - dist_to_ex1])
+    preds = np.array([1. - ( dist_to_ex0 / HYPERDIMS ), 1 - ( dist_to_ex1 / HYPERDIMS )])
+    print(preds)
     # preds = np.array([-1. * dist_to_ex0, -1. * dist_to_ex1])
     preds = np.exp(preds)/sum(np.exp(preds))
     # preds = preds / preds.sum()