Upload 6 files
Browse files- Dockerfile +25 -0
- README.md +5 -6
- app.py +34 -0
- gitattributes +35 -0
- input.jpg +0 -0
- requirements.txt +2 -0
Dockerfile
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM registry.us-west-1.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda11.8.0-py310-torch2.1.0-tf2.14.0-1.10.0
|
2 |
+
ENV DEBIAN_FRONTEND=noninteractive
|
3 |
+
|
4 |
+
RUN useradd -m -u 1000 user
|
5 |
+
USER user
|
6 |
+
ENV HOME=/home/user \
|
7 |
+
PATH=/home/user/.local/bin:${PATH}
|
8 |
+
|
9 |
+
WORKDIR ${HOME}/app
|
10 |
+
|
11 |
+
COPY --chown=1000 ./requirements.txt /tmp/requirements.txt
|
12 |
+
RUN pip install -r /tmp/requirements.txt
|
13 |
+
|
14 |
+
COPY --chown=1000 . ${HOME}/app
|
15 |
+
|
16 |
+
ENV MODELSCOPE_CACHE=${HOME}/cache \
|
17 |
+
PYTHONPATH=${HOME}/app \
|
18 |
+
PYTHONUNBUFFERED=1 \
|
19 |
+
GRADIO_ALLOW_FLAGGING=never \
|
20 |
+
GRADIO_NUM_PORTS=1 \
|
21 |
+
GRADIO_SERVER_NAME=0.0.0.0 \
|
22 |
+
GRADIO_THEME=huggingface \
|
23 |
+
SYSTEM=spaces
|
24 |
+
RUN mkdir -p /home/user/.cache/modelscope/modelscope_modules/ && touch /home/user/.cache/modelscope/modelscope_modules/__init__.py
|
25 |
+
CMD ["python", "app.py"]
|
README.md
CHANGED
@@ -1,12 +1,11 @@
|
|
1 |
---
|
2 |
-
title:
|
3 |
-
emoji:
|
4 |
-
colorFrom:
|
5 |
colorTo: indigo
|
6 |
-
sdk:
|
7 |
-
sdk_version: 4.26.0
|
8 |
-
app_file: app.py
|
9 |
pinned: false
|
|
|
10 |
---
|
11 |
|
12 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
1 |
---
|
2 |
+
title: Old Photo Restoration
|
3 |
+
emoji: π
|
4 |
+
colorFrom: yellow
|
5 |
colorTo: indigo
|
6 |
+
sdk: docker
|
|
|
|
|
7 |
pinned: false
|
8 |
+
license: apache-2.0
|
9 |
---
|
10 |
|
11 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
app.py
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import os
|
3 |
+
import cv2
|
4 |
+
from modelscope.outputs import OutputKeys
|
5 |
+
from modelscope.pipelines import pipeline
|
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'
|
14 |
+
##result = img_colorization(img_path)
|
15 |
+
##cv2.imwrite('result.png', result[OutputKeys.OUTPUT_IMG])
|
16 |
+
def color(image):
|
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)
|
gitattributes
ADDED
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
input.jpg
ADDED
requirements.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
gradio
|
2 |
+
gradio_imageslider
|