34567ihcxsdrtyuiop

app.py

@@ -21,13 +21,16 @@ def load_model(model_path, device):
     model.eval()  # Set model to evaluation mode
     return model
 
+from concrete.fhe import Configuration
 
 def load_secure_model(model):
     print("Compiling secure model...")
     secure_model = compile_torch_model(
         model.to("cpu"), 
         n_bits={"model_inputs": 4, "op_inputs": 3, "op_weights": 3, "model_outputs": 5},
-        # rounding_threshold_bits={"n_bits": 7, "method": "APPROXIMATE"},
+        rounding_threshold_bits={"n_bits": 7, "method": "APPROXIMATE"},
+        p_error=0.05,
+        configuration=Configuration(enable_tlu_fusing=True, print_tlu_fusing=False, use_gpu=False),
         torch_inputset=torch.rand(10, 3, 224, 224)
     )
     return secure_model