Khrisna commited on
Commit
45b4d1a
·
1 Parent(s): 9f8372e

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +7 -3
index.js CHANGED
@@ -134,10 +134,14 @@ app.post('/api/waifu2x', async (req, res) => {
134
  const response = await waifu2x(data_img, format)
135
  res.setHeader('Content-Type', 'image/jpg')
136
  res.send(response)
 
 
 
 
137
  } else {
138
- const response = await waifu2x(Buffer.from(images, "base64", format)
139
- res.setHeader('Content-Type', 'image/jpg')
140
- res.send(response)
141
  }
142
  } catch (e) {
143
  console.log(e)
 
134
  const response = await waifu2x(data_img, format)
135
  res.setHeader('Content-Type', 'image/jpg')
136
  res.send(response)
137
+ } else if (images && typeof images == 'string' && isBase64(images)) {
138
+ const response = await waifu2x(Buffer.from(images, "base64"), format)
139
+ res.setHeader('Content-Type', 'image/jpg')
140
+ res.send(response)
141
  } else {
142
+ res.json({
143
+ success: false, message: 'No url or base64 detected!!'
144
+ })
145
  }
146
  } catch (e) {
147
  console.log(e)