replace upload image by url input

S2I/modules/utils.py

@@ -85,8 +85,7 @@ def get_s2i_home() -> str:
 def download_models():
     urls = {
         '350k-adapter': 'https://huggingface.co/myn0908/sk2ks/resolve/main/adapter_weights_large_sketch2image_lora.pkl?download=true',
-        '350k': 'https://huggingface.co/myn0908/sk2ks/resolve/main/sketch_to_image_mixed_weights_350k_lora.pkl?download=true',
-        '100k': 'https://huggingface.co/myn0908/sk2ks/resolve/main/model_16001.pkl?download=true',
+        '350k': 'https://huggingface.co/myn0908/sk2ks/resolve/main/sketch_to_image_mixed_weights_350k_lora.pkl?download=true'
     }
     # Get the current working directory
     home = get_s2i_home()

app.py

@@ -82,11 +82,10 @@ def convert_to_pencil_sketch(image):
 def get_meta_from_image(input_img, type_image):
     if input_img is None:
         return gr.update(value=None)
+        
+    img = Image.open(BytesIO(requests.get(input_img).content)).convert('RGB')
 
-    file_content, _ = read_temp_file(input_img)
-    
     # Read the image using Pillow
-    img = Image.open(io.BytesIO(file_content)).convert("RGB")
     img_np = np.array(img)
 
     if type_image == 'RGB':
@@ -97,10 +96,32 @@ def get_meta_from_image(input_img, type_image):
 
     # Convert the processed image back to PIL Image
     img_pil = Image.fromarray(processed_img.astype('uint8'))
-    
+
     return img_pil
 
 
+# def get_meta_from_image(input_img, type_image):
+#     if input_img is None:
+#         return gr.update(value=None)
+
+#     file_content, _ = read_temp_file(input_img)
+    
+#     # Read the image using Pillow
+#     img = Image.open(io.BytesIO(file_content)).convert("RGB")
+#     img_np = np.array(img)
+
+#     if type_image == 'RGB':
+#         sketch = convert_to_pencil_sketch(img_np)
+#         processed_img = 255 - sketch
+#     elif type_image == 'SKETCH':
+#         processed_img = 255 - img_np
+
+#     # Convert the processed image back to PIL Image
+#     img_pil = Image.fromarray(processed_img.astype('uint8'))
+    
+#     return img_pil
+
+
 
 with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
     gr.HTML(
@@ -200,7 +221,8 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
                 canvas_size=(1024, 1024),
                 layers=False
             )
-            input_image = gr.File(label='Input image')
+            # input_image = gr.File(label='Input image')
+            url_image = gr.Textbox(label="Image URLS", value="")
 
             download_sketch = gr.Button(
                 "Download sketch", scale=1, elem_id="download_sketch"
@@ -263,8 +285,8 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
                                         label="Demo Speed",
                                         interactive=True)
                                 model_options = gr.Radio(
-                                        choices=["100k", "350k", "350k-adapter"],
-                                        value="350k",
+                                        choices=["350k", "350k-adapter"],
+                                        value="350k-adapter",
                                         label="Type Sketch2Image models",
                                         interactive=True)
                         
@@ -291,15 +313,15 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
     outputs = [result, download_sketch]
     prompt.submit(fn=assign_gpu, inputs=inputs, outputs=outputs, api_name=False)
 
-    input_image.change(
-                fn=get_meta_from_image,
-                inputs=[
-                    input_image, type_image
-                ],
-                outputs=[
-                    image
-                ]
-            )
+    # input_image.change(
+    #             fn=get_meta_from_image,
+    #             inputs=[
+    #                 input_image, type_image
+    #             ],
+    #             outputs=[
+    #                 image
+    #             ]
+    #         )
 
     style.change(
         lambda x: controller.styles[x],
@@ -322,7 +344,7 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
     val_r.change(assign_gpu, inputs=inputs, outputs=outputs, queue=False, api_name=False)
     run_button.click(fn=assign_gpu, inputs=inputs, outputs=outputs, api_name=False)
     image.change(assign_gpu, inputs=inputs, outputs=outputs, queue=False, api_name=False)
-
+    url_image.submit(fn=get_meta_from_image, inputs=[url_image, type_image], outputs=[image])
 if __name__ == '__main__':
     demo.queue()
     demo.launch(debug=True, share=False)