No image writing

app.py

@@ -1,3 +1,37 @@
+
+Hugging Face's logo Hugging Face
+
+Models
+Datasets
+Spaces
+Posts
+Docs
+Pricing
+
+Spaces:
+Fabrice-TIERCELIN
+/
+Instruct-Pix2Pix-dev
+private
+App
+Files
+Community
+Settings
+Instruct-Pix2Pix-dev
+/ app.py
+Fabrice-TIERCELIN's picture
+Fabrice-TIERCELIN
+No image writing
+4f262b3
+verified
+about 1 hour ago
+raw
+history
+blame
+edit
+delete
+No virus
+10.8 kB
 from diffusers import (
     ControlNetModel,
     DiffusionPipeline,
@@ -24,7 +58,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
 pipe = pipe.to(device)
 
 def check(
-    source_img,
+    input_image,
     prompt,
     negative_prompt,
     denoising_steps,
@@ -34,14 +68,14 @@ def check(
     randomize_seed,
     seed,
     progress = gr.Progress()):
-    if source_img is None:
+    if input_image is None:
         raise gr.Error("Please provide an image.")
 
     if prompt is None or prompt == "":
         raise gr.Error("Please provide a prompt input.")
 
 def pix2pix(
-    source_img,
+    input_image,
     prompt,
     negative_prompt,
     denoising_steps,
@@ -52,7 +86,7 @@ def pix2pix(
     seed,
     progress = gr.Progress()):
     check(
-        source_img,
+        input_image,
         prompt,
         negative_prompt,
         denoising_steps,
@@ -86,17 +120,6 @@ def pix2pix(
     random.seed(seed)
     #pipe = pipe.manual_seed(seed)
 
-    try:
-        imageio.imwrite("data.png", source_img)
-    except:
-        raise gr.Error("Can't read input image. You can try to first save your image in another format (.webp, .png, .jpeg, .bmp...).")
-
-    # Input image
-    try:
-        input_image = Image.open("data.png").convert("RGB")
-    except:
-        raise gr.Error("Can't open input image. You can try to first save your image in another format (.webp, .png, .jpeg, .bmp...).")
-
     original_height, original_width, dummy_channel = np.array(input_image).shape
     output_width = original_width
     output_height = original_height
@@ -172,7 +195,7 @@ with gr.Blocks() as interface:
         """
     )
     with gr.Column():
-        source_img = gr.Image(label = "Your image", sources = ["upload"], type = "numpy")
+        input_image = gr.Image(label = "Your image", sources = ["upload"], type = "pil")
         prompt = gr.Textbox(label = 'Prompt', info = "Instruct what to change in the image", placeholder = 'Order the AI what to change in the image')
         with gr.Accordion("Advanced options", open = False):
             negative_prompt = gr.Textbox(label = 'Negative prompt', placeholder = 'Describe what you do NOT want to see in the image', value = 'Watermark')
@@ -189,7 +212,7 @@ with gr.Blocks() as interface:
         information = gr.Label(label = "Information")
 
     submit.click(check, inputs = [
-        source_img,
+        input_image,
         prompt,
         negative_prompt,
         denoising_steps,
@@ -199,7 +222,7 @@ with gr.Blocks() as interface:
         randomize_seed,
         seed
     ], outputs = [], queue = False, show_progress = False).success(pix2pix, inputs = [
-        source_img,
+        input_image,
         prompt,
         negative_prompt,
         denoising_steps,
@@ -215,7 +238,7 @@ with gr.Blocks() as interface:
 
     gr.Examples(
 	    inputs = [
-            source_img,
+            input_image,
             prompt,
             negative_prompt,
             denoising_steps,
@@ -289,4 +312,4 @@ with gr.Blocks() as interface:
         cache_examples = False,
     )
 
-    interface.queue().launch()
+    interface.queue().launch()