Update app.py

app.py

@@ -18,7 +18,7 @@ pipe = CosmosTextToImagePipeline.from_pretrained(model_id, torch_dtype=torch.bfl
 pipe.to("cuda")
 
 @spaces.GPU
-def generate_image(prompt, negative_prompt, seed, randomize_seed):
+def generate_image(prompt, negative_prompt, seed, randomize_seed, progress=gr.Progress(track_tqdm=True)):
     """
     Generates an image based on the provided prompt and negative prompt.
     Randomizes the seed if randomize_seed is True, otherwise uses the provided seed.
@@ -39,7 +39,7 @@ def generate_image(prompt, negative_prompt, seed, randomize_seed):
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-        # Cosmos Text-to-Image Generator
+        # Cosmos Predict-2 14B Text-to-Image Generator
         Enter a detailed prompt to generate an image using the Cosmos model.
         You can also provide a negative prompt to guide the generation away from certain elements.
         """