Update app.py

app.py

@@ -21,12 +21,12 @@ else:
 
 if torch.cuda.is_available():
     torch.cuda.max_memory_allocated(device=device)
-    pipe = StableDiffusionImg2ImgPipeline.from_pretrained("Envvi/Inkpunk-Diffusion", torch_dtype=torch_dtype)
+    pipe = StableDiffusionImg2ImgPipeline.from_pretrained("Envvi/Inkpunk-Diffusion", torch_dtype=torch_dtype, safety_checker = None, requires_safety_checker = False)
     #pipe = DiffusionPipeline.from_pretrained("Envvi/Inkpunk-Diffusion", torch_dtype=torch.float16, variant="fp16")
     pipe.enable_xformers_memory_efficient_attention()
     pipe = pipe.to(device)
 else: 
-    pipe = StableDiffusionImg2ImgPipeline.from_pretrained("Envvi/Inkpunk-Diffusion", torch_dtype=torch_dtype)
+    pipe = StableDiffusionImg2ImgPipeline.from_pretrained("Envvi/Inkpunk-Diffusion", torch_dtype=torch_dtype, safety_checker = None, requires_safety_checker = False)
     pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -79,7 +79,7 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
-        # Text-to-Image Gradio Template
+        # Image-to-Image Demo
         Currently running on {power_device}.
         """)
 
@@ -110,15 +110,16 @@ with gr.Blocks(css=css) as demo:
                 visible=False,
             )
             
-            seed = gr.Slider(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=0,
-            )
+            with gr.Row():
+                    seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=0,
+                )
             
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
             
             with gr.Row():
                 
@@ -145,7 +146,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=0.0,
                     maximum=10.0,
                     step=0.1,
-                    value=0.75,
+                    value=5.0,
                 )
                 
                 strength = gr.Slider(
@@ -153,7 +154,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=0.0,
                     maximum=1.0,
                     step=0.1,
-                    value=0.75,
+                    value=0.5,
                 )
         
         gr.Examples(