John Smith commited on
Commit
e6ac0b5
·
1 Parent(s): 56d3554

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,9 +7,9 @@ import numpy as np
7
  model = load_model('eee.keras')
8
 
9
  def image_mod(image):
10
- raw_image = image.read()
11
- with open("temporary_image.jpg", "wb") as t:
12
- t.write(raw_image)
13
 
14
  img = cv2.imread("temporary_image.jpg")
15
  resize = tf.image.resize(img, (256, 256))
 
7
  model = load_model('eee.keras')
8
 
9
  def image_mod(image):
10
+ raw_image = image[0].astype(np.uint8)
11
+ temp_image = "temporary_image.jpg"
12
+ cv2.imwrite(temp_image, raw_image)
13
 
14
  img = cv2.imread("temporary_image.jpg")
15
  resize = tf.image.resize(img, (256, 256))