Update index.js

index.js

@@ -16,6 +16,7 @@ const app = express()
 app.set('json spaces', 4)
 app.use(morgan('dev'))
 app.use(express.json())
+app.use(load_model())
 
 app.all('/', (req, res) => {
 	const status = {}
@@ -215,7 +216,6 @@ app.post('/api/nsfw-check', async (req, res) => {
 const PORT = process.env.PORT || 7860
 app.listen(PORT, () => {
   console.log('App running on port', PORT)
-  load_model()
 })
 
 function formatSize(num) {
@@ -340,7 +340,7 @@ async function waifu2x(image, formats) {
     return images.data
 }
 async function check_nsfw(buffer) {
-  let _model
+  let _model = load_model()
   const convert = async (img) => {
   // Decoded image in UInt8 Byte array
     const image = await jpegjs.decode(img, { useTArray: true })
@@ -368,5 +368,5 @@ async function check_nsfw(buffer) {
   return results
 }
 async function load_model() {
-  _model = await nsfwjs.load()
+  return await nsfwjs.load()
 }