Update prompt

app.py

@@ -63,6 +63,25 @@ with image_blocks as demo:
                         rounded=(False, True, True, False),
                         full_width=False,
                     )
+
+                    gr.Markdown("## Image Examples")
+                    gr.Examples(
+                        examples=[os.path.join(os.path.dirname(__file__),
+                                               "examples/exciting-small-kitchen-ideas-1821197-hero-d00f516e2fbb4dcabb076ee9685e877a.jpg"),
+                                  os.path.join(os.path.dirname(__file__),
+                                               "examples/oct-2019-idh-bathroom-reno-ideas-new-gallery-2.jpg"),
+                                  os.path.join(os.path.dirname(__file__),
+                                               "examples/tips-for-decorating-a-beautiful-bedroom-1976169-hero-e960fbb8311c4b9b875a1813962d34eb.jpg"),
+                                  os.path.join(os.path.dirname(__file__),
+                                               "examples/backyard_additions.jpg"),
+                                  os.path.join(os.path.dirname(__file__),
+                                               "examples/living-room-gallery-shelves-l-shaped-couch-ELeyNpyyqpZ8hosOG3EG1X-b5a39646574544e8a75f2961332cd89a.jpg"),
+                                  os.path.join(os.path.dirname(__file__),
+                                               "examples/modern-dining-room-ideas-4147451-hero-d6333998f8b34620adfd4d99ac732586.jpg"),
+                                  ],
+                        inputs=image
+                    )
+
                 with gr.Column():
                     mask_image = gr.Image(label="Mask image", elem_id="mask-img", type="pil").style(height=500)
                     image_out_1 = gr.Image(label="Output 1", elem_id="output-img-1", type="pil").style(height=512)
@@ -72,25 +91,7 @@ with image_blocks as demo:
                 btn.click(fn=predict, inputs=[image, style, backyard_style, color_preference, additional_info],
                           outputs=[image_out_1, image_out_2, image_out_3, mask_image])
 
-            gr.Markdown("## Image Examples")
-            gr.Examples(
-                examples=[os.path.join(os.path.dirname(__file__),
-                                       "examples/exciting-small-kitchen-ideas-1821197-hero-d00f516e2fbb4dcabb076ee9685e877a.jpg"),
-                          os.path.join(os.path.dirname(__file__),
-                                       "examples/modern-kitchen-cabinets.jpg"),
-                          os.path.join(os.path.dirname(__file__),
-                                       "examples/oct-2019-idh-bathroom-reno-ideas-new-gallery-2.jpg"),
-                          os.path.join(os.path.dirname(__file__),
-                                       "examples/tips-for-decorating-a-beautiful-bedroom-1976169-hero-e960fbb8311c4b9b875a1813962d34eb.jpg"),
-                          os.path.join(os.path.dirname(__file__),
-                                       "examples/backyard_additions.jpg"),
-                          os.path.join(os.path.dirname(__file__),
-                                       "examples/living-room-gallery-shelves-l-shaped-couch-ELeyNpyyqpZ8hosOG3EG1X-b5a39646574544e8a75f2961332cd89a.jpg"),
-                          os.path.join(os.path.dirname(__file__),
-                                       "examples/modern-dining-room-ideas-4147451-hero-d6333998f8b34620adfd4d99ac732586.jpg"),
-                          ],
-                inputs=image
-            )
+
 
 image_blocks.queue(concurrency_count=3)
 image_blocks.launch()

model.py

@@ -143,9 +143,9 @@ class VirtualStagingToolV2():
         items = ', '.join(items)
 
         if room == 'backyard':
-            prompt = f'Photorealistic, high resolution, {room} with {style}'
+            prompt = f'Realistic, high resolution, {room} with {style}'
         else:
-            prompt = f'Photorealistic {items}, high resolution, in the {style} style {room}'
+            prompt = f'Realistic {items}, high resolution, in the {style} style {room}'
 
         if color_preference:
             prompt = f"{prompt} in {color_preference}"