Update README.md

README.md

@@ -9,8 +9,10 @@ tags:
 - dreambooth-hackathon
 - bedroom
 widget:
-- text: a photo of room bed, wall, table, curtains in the Acropolis, based on the
+- text: >-
+    a photo of room bed, wall, table, curtains in the Acropolis, based on the
     input image
+pipeline_tag: image-to-text
 ---
 
 # DreamBooth model for the room concept trained by gaurav761 on the gaurav761/bedroom-image dataset.
@@ -33,4 +35,4 @@ from diffusers import StableDiffusionPipeline
 pipeline = StableDiffusionPipeline.from_pretrained('gaurav761/bedroom5')
 image = pipeline().images[0]
 image
-```
+```