Update app.py

app.py

@@ -38,6 +38,7 @@ def generate_caption(image):
         image_size=(image.width, image.height)
     )
     prompt =  parsed_answer["<MORE_DETAILED_CAPTION>"]
+    print(f'\033[1mGeneration} completed!\033[0m ({prompt})')
     yield prompt, None
     image_path = generate_image(prompt,random.randint(0, 4294967296))
     yield prompt, image_path