Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -32,10 +32,11 @@ def read_image_file(file) -> Image.Image:
|
|
| 32 |
|
| 33 |
def pil_to_base64(img, format="jpeg", encode=False):
|
| 34 |
if encode:
|
| 35 |
-
|
| 36 |
-
img.save(
|
| 37 |
-
|
| 38 |
-
|
|
|
|
| 39 |
else:
|
| 40 |
temp_path = f"temp.{format}"
|
| 41 |
img.save(temp_path)
|
|
|
|
| 32 |
|
| 33 |
def pil_to_base64(img, format="jpeg", encode=False):
|
| 34 |
if encode:
|
| 35 |
+
bytes_object = BytesIO()
|
| 36 |
+
img.save(bytes_object, format)
|
| 37 |
+
bytes_object.seek(0)
|
| 38 |
+
b64 = base64.b64encode(bytes_object.getvalue()).decode("ascii")
|
| 39 |
+
return b64
|
| 40 |
else:
|
| 41 |
temp_path = f"temp.{format}"
|
| 42 |
img.save(temp_path)
|