WenqingZhang commited on
Commit
490db96
·
verified ·
1 Parent(s): ad78cf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -111,8 +111,7 @@ def encode_quantize_encrypt(text, user_id):
111
  else:
112
  fhe_api = FHEModelClient(FHE_MODEL_PATH, f".fhe_keys/{user_id}")
113
  encodings = transformer_vectorizer.transform([text])
114
-
115
- print(encodings)
116
  fhe_api.load()
117
  quantized_encodings = fhe_api.model.quantize_input(encodings).astype(numpy.uint8)
118
  encrypted_quantized_encoding = fhe_api.quantize_encrypt_serialize(encodings)
 
111
  else:
112
  fhe_api = FHEModelClient(FHE_MODEL_PATH, f".fhe_keys/{user_id}")
113
  encodings = transformer_vectorizer.transform([text])
114
+ print("Encodings shape:", encodings.shape)
 
115
  fhe_api.load()
116
  quantized_encodings = fhe_api.model.quantize_input(encodings).astype(numpy.uint8)
117
  encrypted_quantized_encoding = fhe_api.quantize_encrypt_serialize(encodings)