Revert

app.py

@@ -18,19 +18,15 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
 # Function to generate an image
-@spaces.GPU
+@spaces.GPU  # Adjust the duration as needed
 def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress=gr.Progress()):
-    pipe.to('cuda')
+    pipe.to('cuda')  # Move the model to GPU when the function is called
     
     if randomize_seed:
         seed = random.randint(0, 99999999)
-    
-    original_prompt = prompt
-    original_negative_prompt = negative_prompt
-    
     if use_defaults:
         prompt = f"{prompt}, masterpiece, best quality"
-        negative_prompt = f"nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name, {negative_prompt}"
+        negative_prompt = f"lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name, {negative_prompt}"
     generator = torch.manual_seed(seed)
     
     def callback(step, timestep, latents):
@@ -52,31 +48,15 @@ def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_s
 
     torch.cuda.empty_cache()
 
-    return image, seed, original_prompt, original_negative_prompt
+    return image, seed
 
 # Define Gradio interface
 def interface_fn(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress=gr.Progress()):
-    image, seed, original_prompt, original_negative_prompt = generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress)
-    
-    details = f"""Prompt: {original_prompt}
-
-Negative prompt: {original_negative_prompt}
-
-Steps: {num_inference_steps}, CFG scale: {guidance_scale}, Seed: {seed}, Size: {resolution}
-
-Default quality tags: {"Enabled" if use_defaults else "Disabled"}"""
-
-    if use_defaults:
-        details += f"""
-
-Default prompt addition: , masterpiece, best quality
-
-Default negative prompt addition: nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name"""
-    
-    return image, seed, gr.update(value=seed), details
+    image, seed = generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress)
+    return image, seed, gr.update(value=seed)
 
 def reset_inputs():
-    return gr.update(value=''), gr.update(value=''), gr.update(value=True), gr.update(value='832x1216'), gr.update(value=7), gr.update(value=28), gr.update(value=0), gr.update(value=True)
+    return gr.update(value=''), gr.update(value='realistic, 3d,'), gr.update(value=True), gr.update(value='832x1216'), gr.update(value=7), gr.update(value=28), gr.update(value=0), gr.update(value=True)
 
 with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as demo:
     gr.HTML(
@@ -86,7 +66,7 @@ with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as d
     with gr.Row():
         with gr.Column():
             prompt_input = gr.Textbox(lines=2, placeholder="Enter prompt here", label="Prompt")
-            negative_prompt_input = gr.Textbox(lines=2, placeholder="Enter negative prompt here", label="Negative Prompt")
+            negative_prompt_input = gr.Textbox(lines=2, placeholder="Enter negative prompt here", label="Negative Prompt", value="realistic, 3d,")
             use_defaults_input = gr.Checkbox(label="Use Default Quality Tags and Negative Prompt", value=True)
             resolution_input = gr.Radio(
                 choices=[
@@ -105,14 +85,6 @@ with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as d
 
         with gr.Column():
             output_image = gr.Image(type="pil", label="Generated Image")
-        with gr.Accordion("Generation Details", open=False):
-            generation_info = gr.Textbox(
-                label="", 
-                max_lines=15, 
-                interactive=False,
-                elem_id="generation_info",
-                show_copy_button=True
-            )
             gr.Markdown(
                 """
                 ### Recommended prompt formatting:
@@ -131,7 +103,7 @@ with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as d
         inputs=[
             prompt_input, negative_prompt_input, use_defaults_input, resolution_input, guidance_scale_input, num_inference_steps_input, seed_input, randomize_seed_input
         ],
-        outputs=[output_image, seed_input, seed_input, generation_info]
+        outputs=[output_image, seed_input]
     )
     
     reset_button.click(