commit

app.py

@@ -132,6 +132,13 @@ def encode_quantize(test_file, eval_key, encodings):
 def encrypt_encoded_quantize(encodings):
     fhe_api = FHEModelClient(f"fhe_model", f".fhe_keys/{eval_key}")
     fhe_api.load()
+
+    encodings = np.array(encodings, dtype=np.float32)
+
+    # Check for NaN or Inf values
+    if np.isnan(encodings).any() or np.isinf(encodings).any():
+        raise ValueError("Encodings contain NaN or Inf values, which cannot be processed.")
+
     quantized_encodings = fhe_api.model.quantize_input(encodings).astype(numpy.uint8)
     encrypted_quantized_encoding = fhe_api.quantize_encrypt_serialize(encodings)