Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -72,7 +72,7 @@ def load_models():
|
|
| 72 |
return pipe
|
| 73 |
|
| 74 |
|
| 75 |
-
def resize_for_condition_image(input_image
|
| 76 |
input_image = input_image.convert("RGB")
|
| 77 |
W, H = input_image.size
|
| 78 |
k = float(resolution) / min(H, W)
|
|
@@ -99,7 +99,7 @@ def generate_qr_code(content: str):
|
|
| 99 |
return img
|
| 100 |
|
| 101 |
|
| 102 |
-
def image_to_base64(image
|
| 103 |
buffered = BytesIO()
|
| 104 |
image.save(buffered, format="PNG")
|
| 105 |
return base64.b64encode(buffered.getvalue()).decode("utf-8")
|
|
|
|
| 72 |
return pipe
|
| 73 |
|
| 74 |
|
| 75 |
+
def resize_for_condition_image(input_image, resolution: int):
|
| 76 |
input_image = input_image.convert("RGB")
|
| 77 |
W, H = input_image.size
|
| 78 |
k = float(resolution) / min(H, W)
|
|
|
|
| 99 |
return img
|
| 100 |
|
| 101 |
|
| 102 |
+
def image_to_base64(image):
|
| 103 |
buffered = BytesIO()
|
| 104 |
image.save(buffered, format="PNG")
|
| 105 |
return base64.b64encode(buffered.getvalue()).decode("utf-8")
|