Duplicate from sergeipetrov/dat-x2-ie

.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

README.md

@@ -0,0 +1,22 @@
+---
+tags:
+- image-to-image
+- super-resolution
+library_name: generic
+---
+
+## DAT-x2 for 🤗Inference endpoint deployment.
+
+
+### Expected payload:
+Curl:
+```json
+curl URL \
+        -X POST \
+        --data-binary @car.png \
+        -H "Content-Type: image/png"
+```
+Python:
+```python
+requests.post(ENDPOINT_URL, headers={"Content-Type": "image/png"}, data=open("car.png", 'rb').read()).json()
+```

handler.py

@@ -0,0 +1,20 @@
+from image_gen_aux import UpscaleWithModel
+import numpy as np
+from PIL import Image
+from io import BytesIO
+import base64
+
+
+class EndpointHandler():
+    def __init__(self, path=""):
+        self.upscaler = UpscaleWithModel.from_pretrained("OzzyGT/DAT_X2").to("cuda")
+        
+    def __call__(self, data):
+        img = data.pop("inputs", data)
+        output = self.upscaler(img, tiling=True, tile_width=768, tile_height=768, overlap=8)
+        
+        buffered = BytesIO()
+        output.save(buffered, format="JPEG")
+        img_str = base64.b64encode(buffered.getvalue())
+
+        return img_str.decode()

requirements.txt

@@ -0,0 +1 @@
+git+https://github.com/asomoza/image_gen_aux.git