Add Replicate demo (#428)

* add cog.yaml * add cog predict * add cog predict * update cog predict * update cog predict * add alpha png * update cog predict * update cog predict * update cog predict * update readme * fix codespell
2022-09-05 00:13:42 +08:00
parent e2db576020
commit e5763af574
7 changed files with 178 additions and 5 deletions
--- a/cog.yaml
+++ b/cog.yaml
@@ -0,0 +1,22 @@
+# This file is used for constructing replicate env
+image: "r8.im/tencentarc/realesrgan"
+
+build:
+  gpu: true
+  python_version: "3.8"
+  system_packages:
+    - "libgl1-mesa-glx"
+    - "libglib2.0-0"
+  python_packages:
+    - "torch==1.7.1"
+    - "torchvision==0.8.2"
+    - "numpy==1.21.1"
+    - "lmdb==1.2.1"
+    - "opencv-python==4.5.3.56"
+    - "PyYAML==5.4.1"
+    - "tqdm==4.62.2"
+    - "yapf==0.31.0"
+    - "basicsr==1.4.2"
+    - "facexlib==0.2.5"
+
+predict: "cog_predict.py:Predictor"