Update app.py

app.py

@@ -92,7 +92,7 @@ def generate_caption_and_image(image, f, p, d):
         num_inference_steps=50,
         max_sequence_length=512,
         generator=torch.Generator("cpu").manual_seed(0)
-        ).images
+        ).images[0]
         return image