Update helper.py

helper.py

@@ -1,7 +1,7 @@
 import io
 from PIL import Image
 import base64
-#Define the Functions
+#Define the ConversionFunctions
 def image_to_base64_str(pil_image):
     byte_arr = io.BytesIO()
     pil_image.save(byte_arr, format='PNG')
@@ -12,3 +12,12 @@ def base64_to_pil(img_base64):
     byte_stream = io.BytesIO(base64_decoded)
     pil_image = Image.open(byte_stream)
     return pil_image
+
+#Define Image Captioner and Image Generate Functions
+def captioner(image):
+    base64_image = image_to_base64_str(image)
+    result = imagecaptioningpipe(base64_image)
+    return result[0]['generated_text']
+
+def generate(prompt):
+    return imagegenpipe(prompt).images[0]