Add nsfwjs

flexbox · flexbox · commit d45e27e9dbef · 2020-07-18T17:07:10.000+02:00
diff --git a/nsfwjs/README.md b/nsfwjs/README.md
@@ -0,0 +1,10 @@
+# [nsfwjs](https://github.com/infinitered/nsfwjs)
+
+> Classify images with `nsfwjs`
+
+## Getting Started
+
+```
+yarn
+yarn start
+```
diff --git a/nsfwjs/index.html b/nsfwjs/index.html
@@ -0,0 +1,23 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <title>TensorFlow Template</title>
+    <meta charset="UTF-8" />
+    <link
+      rel="stylesheet"
+      href="./node_modules/modern-normalize/modern-normalize.css"
+    />
+  </head>
+
+  <body>
+    <div id="app"></div>
+    <h2 id="message">Detecting image…</h2>
+    <img
+      id="image"
+      crossorigin
+      src="https://images.unsplash.com/photo-1492632503949-a31d9972b400?ixlib=rb-1.2.1&ixid=eyJhcHBfaWQiOjEyMDd9&auto=format&fit=crop&w=1267&q=80"
+    />
+
+    <script src="./src/index.js"></script>
+  </body>
+</html>
diff --git a/nsfwjs/package.json b/nsfwjs/package.json
@@ -0,0 +1,20 @@
+{
+  "name": "tensorflowjs-template",
+  "version": "1.0.0",
+  "description": "",
+  "main": "index.html",
+  "scripts": {
+    "start": "parcel index.html --open",
+    "build": "parcel build index.html"
+  },
+  "dependencies": {
+    "@tensorflow/tfjs": "^2.0.1",
+    "modern-normalize": "^0.7.0",
+    "nsfwjs": "^2.2.0"
+  },
+  "devDependencies": {
+    "@babel/core": "7.2.0",
+    "parcel-bundler": "^1.6.1"
+  },
+  "keywords": []
+}
diff --git a/nsfwjs/src/index.js b/nsfwjs/src/index.js
@@ -0,0 +1,27 @@
+import * as tf from '@tensorflow/tfjs';
+import * as nsfwjs from 'nsfwjs';
+import './styles.css';
+
+document.getElementById('app').innerHTML = `
+<code>TensorFlow version: ${tf.version.tfjs}</code>
+`;
+
+const targetImage = document.getElementById('image');
+const targetText = document.getElementById('message');
+
+// Load model from my S3.
+// See the section hosting the model files on your site.
+nsfwjs.load().then(function (model) {
+  model.classify(targetImage).then(function (predictions) {
+    // Classify the image
+    targetText.innerHTML = `
+      <ul>
+        <li>${predictions[0].className}: ${predictions[0].probability}</li>
+        <li>${predictions[1].className}: ${predictions[1].probability}</li>
+        <li>${predictions[2].className}: ${predictions[2].probability}</li>
+        <li>${predictions[3].className}: ${predictions[3].probability}</li>
+        <li>${predictions[4].className}: ${predictions[4].probability}</li>
+      </ul>
+  `;
+  });
+});
diff --git a/nsfwjs/src/styles.css b/nsfwjs/src/styles.css
@@ -0,0 +1,3 @@
+body {
+  padding: 20px 40px;
+}
diff --git a/nsfwjs/yarn.lock b/nsfwjs/yarn.lock

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+body {`
	`2`	`+ padding: 20px 40px;`
	`3`	`+}`