Spaces:
Configuration error
Configuration error
Danila-Pechenev
commited on
Commit
•
947bd52
1
Parent(s):
9bd7d18
Cosmetics
Browse files- 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 |
-
|
|
|
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
|