Upload 6 files

Dockerfile

@@ -0,0 +1,25 @@
+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
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:${PATH}
+
+WORKDIR ${HOME}/app
+
+COPY --chown=1000 ./requirements.txt /tmp/requirements.txt
+RUN pip install -r /tmp/requirements.txt
+
+COPY --chown=1000 . ${HOME}/app
+
+ENV MODELSCOPE_CACHE=${HOME}/cache \
+    PYTHONPATH=${HOME}/app \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces
+RUN mkdir -p /home/user/.cache/modelscope/modelscope_modules/ && touch /home/user/.cache/modelscope/modelscope_modules/__init__.py
+CMD ["python", "app.py"]

README.md

@@ -1,12 +1,11 @@
 ---
-title: Old2new
-emoji: 👀
-colorFrom: gray
+title: Old Photo Restoration
+emoji: 🌖
+colorFrom: yellow
 colorTo: indigo
-sdk: gradio
-sdk_version: 4.26.0
-app_file: app.py
+sdk: docker
 pinned: false
+license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,34 @@
+import gradio as gr
+import os
+import cv2
+from modelscope.outputs import OutputKeys
+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'
+##result = img_colorization(img_path)
+##cv2.imwrite('result.png', result[OutputKeys.OUTPUT_IMG])
+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 (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=ImageSlider(position=0.5,label='Colored image with slider-view'),examples=examples,title=title,description=description)
+
+if __name__ == "__main__":
+    demo.launch(share=False)

gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+gradio_imageslider