simdi commited on
Commit
d33db96
·
verified ·
1 Parent(s): 9b34f86

Update handler.py

Browse files
Files changed (1) hide show
  1. 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: Image.Image, resolution: int):
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: Image.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")