mariotawfik commited on
Commit
fb70103
·
1 Parent(s): 21c5249
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -148,7 +148,7 @@ def encode_quantize(test_file, eval_key, encodings):
148
 
149
  encodings = np.array(encodings).reshape(1, -1)
150
 
151
- return encodings
152
 
153
  def encrypt_encoded_quantize(encodings):
154
  fhe_api = FHEModelClient(f"fhe_model", f".fhe_keys/{eval_key}")
 
148
 
149
  encodings = np.array(encodings).reshape(1, -1)
150
 
151
+ return encodings[0]
152
 
153
  def encrypt_encoded_quantize(encodings):
154
  fhe_api = FHEModelClient(f"fhe_model", f".fhe_keys/{eval_key}")