Update app.py

app.py

@@ -21,6 +21,7 @@ pipe = DiffusionPipeline.from_pretrained(
     use_auth_token=auth_token,
     torch_dtype=dtype,
     low_cpu_mem_usage=True
+    
 )
   # Make sure your HF token has gated access to Flux