wenjun99 commited on
Commit
ff2b00a
·
verified ·
1 Parent(s): 05d848e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -34,6 +34,16 @@ def string_to_binary_labels(s: str) -> list[int]:
34
  bits.extend(char_bits)
35
  return bits
36
 
 
 
 
 
 
 
 
 
 
 
37
  def clean_image(img: Image.Image, min_size: int = 256) -> Image.Image:
38
  img = img.convert("RGB")
39
  if img.width < min_size or img.height < min_size:
 
34
  bits.extend(char_bits)
35
  return bits
36
 
37
+ def binary_labels_to_string(bits: list[int]) -> str:
38
+ chars = []
39
+ for i in range(0, len(bits), 8):
40
+ byte = bits[i:i+8]
41
+ if len(byte) < 8:
42
+ byte += [0] * (8 - len(byte))
43
+ ascii_val = sum(b << (7 - j) for j, b in enumerate(byte))
44
+ chars.append(chr(ascii_val))
45
+ return ''.join(chars)
46
+
47
  def clean_image(img: Image.Image, min_size: int = 256) -> Image.Image:
48
  img = img.convert("RGB")
49
  if img.width < min_size or img.height < min_size: