Update app.py

app.py

@@ -9,7 +9,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.flo
 pipe = pipe.to("cuda")
 
 @spaces.GPU()
-def text_to_image(prompt, negative_prompt, steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
+def text_to_image(prompt, negative_prompt, steps, guidance_scale, add_4k_masterpiece, progress=gr.Progress(track_tqdm=True)):
     if add_4k_masterpiece:
         prompt += ", 4k, (masterpiece)"
     image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=steps, guidance_scale=guidance_scale).images[0]