commit

app.py

@@ -148,12 +148,14 @@ def encode_quantize(test_file, eval_key, encodings):
 
     encodings = np.array(encodings).reshape(1, -1)
 
-    return encodings[0]
+    return encodings
 
 def encrypt_encoded_quantize(encodings):
     fhe_api = FHEModelClient(f"fhe_model", f".fhe_keys/{eval_key}")
     fhe_api.load()
 
+    encodings = encodings[0]
+    
     if isinstance(encodings, str):
         encodings = safe_eval(encodings)