tommy24 commited on
Commit
bb811a7
·
1 Parent(s): ad6a137

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -764,7 +764,7 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
764
  def encode_image(image: Image.Image, format: str = "PNG") -> str:
765
  with io.BytesIO() as buffer:
766
  image.save(buffer, format=format)
767
- encoded_image = base64.b64encode(buffer.getvalue()).decode("utf-8")
768
  return encoded_image
769
 
770
  def url_to_base64(image_url):
@@ -795,7 +795,7 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
795
  # base64_image_str = encode_image(image)
796
 
797
  if image_data:
798
- with io.BytesIO(base64.b64decode(image_data)) as buffer:
799
  image = Image.open(buffer)
800
  base64_image_str = encode_image(image)
801
 
 
764
  def encode_image(image: Image.Image, format: str = "PNG") -> str:
765
  with io.BytesIO() as buffer:
766
  image.save(buffer, format=format)
767
+ encoded_image = buffer.getvalue().decode("latin-1") # Use "latin-1" encoding for bytes to str conversion
768
  return encoded_image
769
 
770
  def url_to_base64(image_url):
 
795
  # base64_image_str = encode_image(image)
796
 
797
  if image_data:
798
+ with io.BytesIO(image_data) as buffer:
799
  image = Image.open(buffer)
800
  base64_image_str = encode_image(image)
801