Update index.js

index.js

@@ -213,7 +213,10 @@ app.post('/api/nsfw-check', async (req, res) => {
 })*/
 
 const PORT = process.env.PORT || 7860
-app.listen(PORT, () => console.log('App running on port', PORT))
+app.listen(PORT, () => {
+  console.log('App running on port', PORT)
+  load_model()
+)}
 
 function formatSize(num) {
 	return bytes(+num || 0, { unitSeparator: ' ' })
@@ -337,7 +340,7 @@ async function waifu2x(image, formats) {
     return images.data
 }
 async function check_nsfw(buffer) {
-  const model = await nsfwjs.load() // To load a local model, nsfw.load('file://./path/to/model/')
+  let _model
   const convert = async (img) => {
   // Decoded image in UInt8 Byte array
     const image = await jpegjs.decode(img, { useTArray: true })
@@ -353,7 +356,7 @@ async function check_nsfw(buffer) {
     return tfjs.tensor3d(values, [image.height, image.width, numChannels], 'int32')
   }
   const image = await convert(buffer)
-  const predictions = await model.classify(image)
+  const predictions = await _model.classify(image)
   image.dispose();
   const results = predictions.map(v => {
     return {
@@ -363,4 +366,7 @@ async function check_nsfw(buffer) {
     }
   })
   return results
+}
+async function load_model() {
+  _model = await nsfwjs.load()
 }