Update app.py

app.py

@@ -1,31 +1,34 @@
 from diffusers import StableDiffusionXLPipeline
 import torch
-from gradio import Interface, Image, Dropdown, Slider
 import gradio as gr
-import spaces
 
+# Load the model with forced download
 model_id = "RunDiffusion/Juggernaut-X-v10"
-pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float32, force_download=True)
+pipe = pipe.to("cpu")
 
-@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):
+    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")
 
-gradio_interface = Interface(
+gradio_interface = gr.Interface(
     fn=text_to_image,
-    allow_duplication=False,
     inputs=[
-        gr.Textbox(label="Prompt", lines=4, placeholder="Enter your prompt here..."),
-        gr.Textbox(label="Negative Prompt", lines=4, placeholder="worst quality, low quality, bad anatomy, bad hands, missing fingers, fewer digits, source_furry, source_pony, source_cartoon,3d, blurry,", ),
+        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.Slider(minimum=1, maximum=20, value=7.5, label="Guidance Scale", step=0.1)
+        gr.Slider(minimum=1, maximum=20, value=7.5, label="Guidance Scale", step=0.1),
+        gr.Checkbox(label="Add recommended prompt items (4k, masterpiece)", value=False)
+    ],
+    outputs=gr.Image(type="pil", show_download_button=True),
+    examples=[
+        [" hyper realistic , detailed , real person , very pretty naked Japanese 18 yo girl, looking at the viewer, ashamed, (smile:0.7), textured fair skin, tight skirt, She is standing up in an open-air bath in a Japanese hot spring, medium breasts , skinny but fit , 8k , Leica quality ,  masterpiece , , 4k, high quality, (masterpiece)", "worst quality, low quality, bad anatomy, bad hands, missing fingers, fewer digits, source_furry, source_pony, source_cartoon,3d, blurry,", 60, 9.5, False],
     ],
-    outputs=Image(type="pil", show_download_button=True),
-    examples= [" hyper realistic , detailed , real person , very pretty naked Japanese 18 yo girl, looking at the viewer, ashamed, (smile:0.7), textured fair skin, tight skirt, She is standing up in an open-air bath in a Japanese hot spring, medium breasts , skinny but fit , 8k , Leica quality ,  masterpiece , "],
     cache_examples=False,
     theme=gr.themes.Soft()
 )
-gradio_interface.launch()
+
+# Launch the Gradio app with share=True
+gradio_interface.launch(share=True)