No image writing

app.py

@@ -19,7 +19,7 @@ def noise_color(color, noise):
     return color + random.randint(- noise, noise)
 
 def check(
-    source_img,
+    input_image,
     enlarge_top,
     enlarge_right,
     enlarge_bottom,
@@ -36,7 +36,7 @@ def check(
     seed,
     debug_mode,
     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 == "":
@@ -63,7 +63,7 @@ def check(
         raise gr.Error("At least one border must be enlarged.")
 
 def uncrop(
-    source_img,
+    input_image,
     enlarge_top,
     enlarge_right,
     enlarge_bottom,
@@ -81,7 +81,7 @@ def uncrop(
     debug_mode,
     progress = gr.Progress()):
     check(
-        source_img,
+        input_image,
         enlarge_top,
         enlarge_right,
         enlarge_bottom,
@@ -140,17 +140,6 @@ def uncrop(
     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, original_channel = np.array(input_image).shape
     output_width = enlarge_left + original_width + enlarge_right
     output_height = enlarge_top + original_height + enlarge_bottom
@@ -291,7 +280,7 @@ with gr.Blocks() as interface:
         <li>To modify <b>anything else</b> on your image, I recommend to use <i>Instruct Pix2Pix</i>.</li>
         </ul>
         <br/>
-        🐌 Slow process... ~1 hour.<br>You can duplicate this space on a free account, it works on CPU.<br/>
+        🐌 Slow process... ~1 hour.<br>You can duplicate this space on a free account, it works on CPU and should also run on CUDA.<br/>
         <a href='https://huggingface.co/spaces/Fabrice-TIERCELIN/Uncrop?duplicate=true'><img src='https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14'></a>
         <br/>
         ⚖️ You can use, modify and share the generated images but not for commercial uses.
@@ -309,7 +298,7 @@ with gr.Blocks() as interface:
         with gr.Column():
             enlarge_left = gr.Number(minimum = 0, value = 64, precision = 0, label = "Enlarge on left ⬅️", info = "in pixels")
         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")
         with gr.Column():
             enlarge_right = gr.Number(minimum = 0, value = 64, precision = 0, label = "Enlarge on right ➡️", info = "in pixels")
     with gr.Row():
@@ -353,7 +342,7 @@ with gr.Blocks() as interface:
         enlarged_image,
         mask_image
     ], queue = False, show_progress = False).then(check, inputs = [
-        source_img,
+        input_image,
         enlarge_top,
         enlarge_right,
         enlarge_bottom,
@@ -371,7 +360,7 @@ with gr.Blocks() as interface:
         debug_mode
     ], outputs = [], queue = False,
                      show_progress = False).success(uncrop, inputs = [
-        source_img,
+        input_image,
         enlarge_top,
         enlarge_right,
         enlarge_bottom,
@@ -397,7 +386,7 @@ with gr.Blocks() as interface:
 
     gr.Examples(
 	    inputs = [
-            source_img,
+            input_image,
             enlarge_top,
             enlarge_right,
             enlarge_bottom,