WenqingZhang commited on
Commit
4d3343d
·
verified ·
1 Parent(s): 34923a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -107,7 +107,7 @@ def encode_quantize_encrypt(text, user_id):
107
  raise gr.Error("You need to generate FHE keys first.")
108
  if "legal_rating" in model_names:
109
  fhe_api = FHEModelClient(FHE_LEGAL_PATH, f".fhe_keys/{user_id}")
110
- encodings =vectorizer.fit_transform(data['clause'])
111
  fhe_api = FHEModelClient(FHE_MODEL_PATH, f".fhe_keys/{user_id}")
112
  fhe_api.load()
113
  encodings = transformer_vectorizer.transform([text])
 
107
  raise gr.Error("You need to generate FHE keys first.")
108
  if "legal_rating" in model_names:
109
  fhe_api = FHEModelClient(FHE_LEGAL_PATH, f".fhe_keys/{user_id}")
110
+ encodings =vectorizer.fit_transform([text])
111
  fhe_api = FHEModelClient(FHE_MODEL_PATH, f".fhe_keys/{user_id}")
112
  fhe_api.load()
113
  encodings = transformer_vectorizer.transform([text])