GovindMalhotra commited on
Commit
d129e85
·
1 Parent(s): 6abb53b

another try to fix ndarray error

Browse files

added cv2 instead of image from PIL

Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
3
  from PIL import Image, ImageChops, ImageEnhance
 
4
 
5
  #Converting the images to ELA format for prediction
6
  def convert_to_ela_image(filename, quality):
@@ -9,7 +10,9 @@ def convert_to_ela_image(filename, quality):
9
  # im = Image.open(filename)
10
  # im1 = im.convert('RGB')
11
  # im1.save(resaved_filename, 'JPEG', quality=quality)
12
- filename.save(resaved_filename, 'JPEG', quality=quality)
 
 
13
  resaved_im = Image.open(resaved_filename)
14
 
15
  ela_im = ImageChops.difference(im, resaved_im)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
  from PIL import Image, ImageChops, ImageEnhance
4
+ import cv2
5
 
6
  #Converting the images to ELA format for prediction
7
  def convert_to_ela_image(filename, quality):
 
10
  # im = Image.open(filename)
11
  # im1 = im.convert('RGB')
12
  # im1.save(resaved_filename, 'JPEG', quality=quality)
13
+ # filename.save(resaved_filename, 'JPEG', quality=quality)
14
+ cv2.imwrite(resaved_filename, filename, [cv2.IMWRITE_JPEG_QUALITY, quality])
15
+ # resaved_im = Image.open(resaved_filename)
16
  resaved_im = Image.open(resaved_filename)
17
 
18
  ela_im = ImageChops.difference(im, resaved_im)