Update app.py

app.py

@@ -10,6 +10,8 @@ pipe = pipe.to("cuda")
 
 @spaces.GPU()
 def text_to_image(prompt, negative_prompt, steps, guidance_scale, 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]
     return image
 duplicate_button = gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
@@ -21,6 +23,7 @@ gradio_interface = Interface(
         gr.Textbox(label="Prompt", lines=2, placeholder="Enter your prompt here..."),
         gr.Textbox(label="Negative Prompt", lines=2, placeholder="What to exclude from the image..."),
         gr.Slider(minimum=1, maximum=65, value=50, label="Steps", step=1),
+        gr.Checkbox(label="Add recommended prompt items (4k, masterpiece)", value=False),
         gr.Slider(minimum=1, maximum=20, value=7.5, label="Guidance Scale", step=0.1)
     ],
     outputs=Image(type="pil", show_download_button=True),