Update app.py

app.py

@@ -6,11 +6,10 @@ hf_api_itt_base = os.environ['HF_API_ITT_BASE']
 
 ##### Helper functions #####
 import io, base64, requests, json
-from PIL import Image
 
-def image_to_base64_str(pil_image):
+def image_to_base64_str(image):
     byte_arr = io.BytesIO()
-    pil_image.save(byte_arr, format='PNG')
+    image.save(byte_arr, format='PNG')
     byte_arr = byte_arr.getvalue()
     return str(base64.b64encode(byte_arr).decode('utf-8'))
 
@@ -28,8 +27,7 @@ def get_completion(inputs, parameters = None, ENDPOINT_URL = hf_api_itt_base):
     return json.loads(response.content.decode("utf-8"))
 
 def captioner(image):
-    pil_image = Image.open(image)
-    base64_image = image_to_base64_str(pil_image)
+    base64_image = image_to_base64_str(image)
     result = get_completion(base64_image)
     return result[0]['generated_text']