Update app.py

app.py

@@ -108,11 +108,11 @@ 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:
+        if encodings.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:
+            encodings = np.hstack((dense_matrix, padding))
+        elif encodings.shape[1] > 1736:
             # 截取前1736列
             encodings = dense_matrix[:, :1736]
     else: