Danila-Pechenev commited on
Commit
947bd52
1 Parent(s): 9bd7d18
Files changed (1) hide show
  1. app/model.py +2 -1
app/model.py CHANGED
@@ -28,5 +28,6 @@ def run_model(image_bytes: io.BytesIO, model: keras.Model) -> Image.Image:
28
  output_image_array: np.ndarray = output_image_array.astype(np.uint8)
29
  output_image: Image.Image = Image.fromarray(output_image_array)
30
  output_image: Image.Image = output_image.resize((width, height))
31
- output_image.save("user_data/output.jpg")
 
32
  return output_image
 
28
  output_image_array: np.ndarray = output_image_array.astype(np.uint8)
29
  output_image: Image.Image = Image.fromarray(output_image_array)
30
  output_image: Image.Image = output_image.resize((width, height))
31
+ # Uncomment if necessary:
32
+ # output_image.save("user_data/output.jpg")
33
  return output_image