Update app.py

app.py

@@ -108,10 +108,17 @@ def encode_quantize_encrypt(text, user_id):
     if "legal_rating" in model_names:
         fhe_api = FHEModelClient(FHE_LEGAL_PATH, f".fhe_keys/{user_id}")
         encodings =vectorizer.fit_transform([text]).toarray()
+        if dense_matrix.shape[1] < 1736:
+            # 在后面填充零
+            padding = np.zeros((1, 1736 - dense_matrix.shape[1]))
+            dense_matrix = np.hstack((dense_matrix, padding))
+        elif dense_matrix.shape[1] > 1736:
+            # 截取前1736列
+            encodings = dense_matrix[:, :1736]
     else:
         fhe_api = FHEModelClient(FHE_MODEL_PATH, f".fhe_keys/{user_id}")
         encodings = transformer_vectorizer.transform([text])
-    print("Encodings shape:", encodings.shape)
+    
     fhe_api.load()
     quantized_encodings = fhe_api.model.quantize_input(encodings).astype(numpy.uint8)
     encrypted_quantized_encoding = fhe_api.quantize_encrypt_serialize(encodings)