Update index.js

index.js

@@ -166,11 +166,13 @@ app.post('/api/upscaler', async (req, res) => {
             responseType: "arraybuffer"
           })
           const response = await processImage(data_img.data, denoise, scale, format, type)
-          //res.setHeader('Content-Type', 'image/jpg')
+          const type_img = await fileType.fromBuffer(response)
+          res.setHeader('Content-Type', type_img.mime)
           res.send(response)
         } else if (images && typeof images == 'string' && isBase64(images)) {
 	   	  const response = await processImage(Buffer.from(images, "base64"), denoise, scale, format, type)
-          //res.setHeader('Content-Type', 'image/jpg')
+          const type_img = await fileType.fromBuffer(response)
+          res.setHeader('Content-Type', type_img.mime)
           res.send(response)
         } else {
           res.json({