Update app.py

app.py

@@ -5,6 +5,10 @@ from PIL import Image
 from diffusers import DiffusionPipeline
 import random
 
+torch.backends.cudnn.deterministic = True
+torch.backends.cudnn.benchmark = False
+torch.backends.cuda.matmul.allow_tf32 = True
+
 # Initialize the base model and specific LoRA
 base_model = "black-forest-labs/FLUX.1-dev"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)