cog-ified

ProGamerGov · Sep 29, 2021 · f604000 · f604000
1 parent bd66447
commit f604000
Show file tree

Hide file tree

Showing 5 changed files with 1,154 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,6 @@
 # neural-dream
+<a href="https://replicate.ai/progamergov/neural-dream"><img src="https://img.shields.io/static/v1?label=Replicate&message=Demo and Docker Image&color=darkgreen" height=20></a>
+
 
 This is a PyTorch implementation of DeepDream. The code is based on [neural-style-pt](https://github.com/ProGamerGov/neural-style-pt).
 

diff --git a/__init__.py b/__init__.py
diff --git a/cog.yaml b/cog.yaml
@@ -0,0 +1,13 @@
+build:
+  python_version: "3.7"
+  gpu: true
+  python_packages:
+    - neural_dream==0.0.4
+    - opencv-python==4.5.3.56
+  run:
+    - "mkdir ./models && neural-dream -download_models -download_path ./models"
+  system_packages:
+    - ffmpeg
+    - libsm6
+    - libxext6
+predict: "predict.py:Predictor"
diff --git a/neural_dream.py b/neural_dream.py
@@ -15,6 +15,7 @@
 from neural_dream.dream_auto import auto_model_mode, auto_mean
 
 import argparse
+
 parser = argparse.ArgumentParser()
 # Basic options
 parser.add_argument("-content_image", help="Content target image", default='examples/inputs/tubingen.jpg')