Update app.py

app.py

@@ -15,6 +15,7 @@ import torch
 from PIL import Image
 from functools import partial
 import io
+from io import BytesIO
 
 
 subprocess.run(shlex.split('pip install wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl'))
@@ -94,9 +95,15 @@ def gen_pos_prompt(text):
   pos_prompt = json.loads(response.get('body').read())['results'][0]['outputText']
   return pos_prompt
 
-def encode_image_to_base64(image_path):
-    with open(image_path, "rb") as image_file:
-        encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
+# def encode_image_to_base64(image_path):
+#     with open(image_path, "rb") as image_file:
+#         encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
+#     return encoded_string
+
+def encode_image_to_base64(image):
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")  # Save the image to the buffer
+    encoded_string = base64.b64encode(buffered.getvalue()).decode('utf-8')
     return encoded_string