Update app.py

app.py

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