Update app.py

app.py

@@ -6,7 +6,16 @@ from diffusers import DiffusionPipeline
 #Create Image Generation Pipeline
 imagegenpipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
 
-from helper import image_to_base64_str, base64_to_pil, captioner, generate
+#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]
+    
+from helper import image_to_base64_str, base64_to_pil
 
 #Build Image Caption and Generate Image Game App.
 import gradio as gr