Spaces:
Running
Running
Update app/utils.py
Browse files- app/utils.py +5 -6
app/utils.py
CHANGED
|
@@ -1,8 +1,7 @@
|
|
| 1 |
-
from PIL import Image
|
| 2 |
-
from io import BytesIO
|
| 3 |
from fastapi import UploadFile
|
|
|
|
|
|
|
| 4 |
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
return image
|
|
|
|
|
|
|
|
|
|
| 1 |
from fastapi import UploadFile
|
| 2 |
+
from PIL import Image
|
| 3 |
+
import io
|
| 4 |
|
| 5 |
+
def read_image(upload_file: UploadFile) -> Image.Image:
|
| 6 |
+
image_bytes = upload_file.file.read()
|
| 7 |
+
return Image.open(io.BytesIO(image_bytes)).convert("RGB")
|
|
|