Update app.py

app.py

@@ -3,11 +3,13 @@ import spaces
 import torch
 from diffusers import Cosmos2TextToImagePipeline, EDMEulerScheduler
 import random
+
 model_2b_id = "nvidia/Cosmos-Predict2-2B-Text2Image"
 
-pipe_2b = Cosmos2TextToImagePipeline.from_pretrained(model_2b_id,
-                                                     torch_dtype=torch.bfloat16
-                                                    )
+pipe_2b = Cosmos2TextToImagePipeline.from_pretrained(
+    model_2b_id,
+    torch_dtype=torch.bfloat16
+)
 pipe_2b.to("cuda")
 
 @spaces.GPU(duration=90)