ysharma HF staff commited on
Commit
5712a94
1 Parent(s): c037767

added gradio_imageslider component as output

Browse files

Gradio_imageslider is a custom component in Gradio that allows output with a slider between two images.

Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -6,6 +6,8 @@ from modelscope.pipelines import pipeline
6
  from modelscope.utils.constant import Tasks
7
  import PIL
8
  import numpy as np
 
 
9
 
10
  img_colorization = pipeline(Tasks.image_colorization, model='iic/cv_ddcolor_image-colorization')
11
  img_path = 'input.png'
@@ -15,15 +17,18 @@ def color(image):
15
  output = img_colorization(image[...,::-1])
16
  result = output[OutputKeys.OUTPUT_IMG].astype(np.uint8)
17
  result = result[...,::-1]
 
 
 
18
  print('infer finished!')
19
- return result
20
-
21
 
22
  title = "old_photo_restoration"
23
  description = "upload old photo, ddcolor image colorization"
24
  examples = [['./input.jpg'],]
25
 
26
- demo = gr.Interface(fn=color,inputs="image",outputs="image",examples=examples,title=title,description=description)
27
 
28
  if __name__ == "__main__":
29
  demo.launch(share=False)
 
6
  from modelscope.utils.constant import Tasks
7
  import PIL
8
  import numpy as np
9
+ import uuid
10
+ from gradio_imageslider import ImageSlider
11
 
12
  img_colorization = pipeline(Tasks.image_colorization, model='iic/cv_ddcolor_image-colorization')
13
  img_path = 'input.png'
 
17
  output = img_colorization(image[...,::-1])
18
  result = output[OutputKeys.OUTPUT_IMG].astype(np.uint8)
19
  result = result[...,::-1]
20
+ # Generate a unique filename using UUID
21
+ unique_imgfilename = str(uuid.uuid4()) + '.png'
22
+ cv2.imwrite(unique_imgfilename, result)
23
  print('infer finished!')
24
+ return (image, unique_imgfilename)
25
+
26
 
27
  title = "old_photo_restoration"
28
  description = "upload old photo, ddcolor image colorization"
29
  examples = [['./input.jpg'],]
30
 
31
+ demo = gr.Interface(fn=color,inputs="image",outputs=ImageSlider(position=0.5,label='Colored image with slider-view'),examples=examples,title=title,description=description)
32
 
33
  if __name__ == "__main__":
34
  demo.launch(share=False)