added gradio_imageslider (#1)

- added gradio_imageslider (c0377675a5ef104c350dbaa1c00bff53a0eda1a5)
- added gradio_imageslider component as output (5712a94b116c4c0a82fba43d9322ef81b79f5438)


Co-authored-by: yuvraj sharma <[email protected]>

app.py

@@ -6,6 +6,8 @@ from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
 import PIL
 import numpy as np
+import uuid
+from gradio_imageslider import ImageSlider
 
 img_colorization = pipeline(Tasks.image_colorization, model='iic/cv_ddcolor_image-colorization')
 img_path = 'input.png'
@@ -15,15 +17,18 @@ def color(image):
     output = img_colorization(image[...,::-1])
     result = output[OutputKeys.OUTPUT_IMG].astype(np.uint8)
     result = result[...,::-1]
+    # Generate a unique filename using UUID
+    unique_imgfilename = str(uuid.uuid4()) + '.png'
+    cv2.imwrite(unique_imgfilename, result)
     print('infer finished!')
-    return result
-
+    return (image, unique_imgfilename)
+    
 
 title = "old_photo_restoration"
 description = "upload old photo, ddcolor image colorization"
 examples = [['./input.jpg'],]
 
-demo = gr.Interface(fn=color,inputs="image",outputs="image",examples=examples,title=title,description=description)
+demo = gr.Interface(fn=color,inputs="image",outputs=ImageSlider(position=0.5,label='Colored image with slider-view'),examples=examples,title=title,description=description)
 
 if __name__ == "__main__":
     demo.launch(share=False)

requirements.txt

@@ -1 +1,2 @@
-gradio==3.29.0
+gradio
+gradio_imageslider