Update app.py

app.py

@@ -72,34 +72,23 @@ def keygen(selected_tasks):
     if "financial_rating" in selected_tasks:
         model_names.append('financial_rating')
 
-    # Let's create a user_id
 
 
+    # Let's create a user_id
     user_id = numpy.random.randint(0, 2**32)
     fhe_api = FHEModelClient(FHE_MODEL_PATH, f".fhe_keys/{user_id}")
     fhe_api.load()
 
 
-        # Generate a fresh key
+    # Generate a fresh key
     fhe_api.generate_private_and_evaluation_keys(force=True)
     evaluation_key = fhe_api.get_serialized_evaluation_keys()
 
-        # Save evaluation_key in a file, since too large to pass through regular Gradio
-        # buttons, https://github.com/gradio-app/gradio/issues/1877
-    eval_key_path = Path(f"tmp/tmp_evaluation_key__{user_id}.npy")
-    numpy.save(eval_key_path, evaluation_key)
-    
-    user_id_leagl = numpy.random.randint(0, 2**32)
-    fhe_api_legal= FHEModelClient(FHE_LEGAL_PATH, f".fhe_keys/{user_id_leagl}")
-    fhe_api_legal.load()
-
-
-        
-    evaluation_keys = []
-    evaluation_keys.append(list(evaluation_key)[:ENCRYPTED_DATA_BROWSER_LIMIT])
-    #evaluation_keys.append(list(evaluation_key_legal)[:ENCRYPTED_DATA_BROWSER_LIMIT])
+    # Save evaluation_key in a file, since too large to pass through regular Gradio
+    # buttons, https://github.com/gradio-app/gradio/issues/1877
+    numpy.save(f"tmp/tmp_evaluation_key_{user_id}.npy", evaluation_key)
 
-    return [list(evaluation_key)[:ENCRYPTED_DATA_BROWSER_LIMIT], [user_id]]
+    return [list(evaluation_key)[:ENCRYPTED_DATA_BROWSER_LIMIT], user_id]
 def encode_quantize_encrypt(text, user_id):
     if not user_id:
         raise gr.Error("You need to generate FHE keys first.")