Update app.py

app.py

@@ -4,7 +4,10 @@ import random
 from diffusers import DiffusionPipeline
 import torch
 
-torch_dtype, device = torch.float16, torch.device("cuda") if torch.cuda.is_available() else torch.float32, torch.device("cpu")
+torch_dtype, device = (
+    (torch.float16, torch.device("cuda")) if torch.cuda.is_available() 
+    else (torch.float32, torch.device("cpu"))
+)
 
 model_repo_id = "black-forest-labs/FLUX.1-dev"
 
@@ -14,11 +17,11 @@ pipe.load_lora_weights("pepper13/flux-anime")
 def infer(prompt, randomize_seed, width, height, guidance_scale, num_inference_steps):
     image = pipe(
         prompt=prompt,
-        guidance_scale=guidance_scale, 
-        num_inference_steps=num_inference_steps, 
-        width=width, 
+        guidance_scale=guidance_scale,
+        num_inference_steps=num_inference_steps,
+        width=width,
         height=height
-    ).images[0] 
+    ).images[0]
     
     return image