unused prompt to inpaint

app.py

@@ -20,6 +20,8 @@ segmodel = load_learner("camvid-512.pkl")
 
 working_size = (512, 512)
 
+default_inpainting_prompt = "watercolor of a leafy pedestrian mall at golden hour with multiracial genderqueer joggers and bicyclists and wheelchair users talking and laughing"
+
 seg_vocabulary = ['Animal', 'Archway', 'Bicyclist', 'Bridge', 'Building', 'Car',
        'CartLuggagePram', 'Child', 'Column_Pole', 'Fence', 'LaneMkgsDriv',
        'LaneMkgsNonDriv', 'Misc_Text', 'MotorcycleScooter', 'OtherMoving',
@@ -40,38 +42,18 @@ def get_seg_mask(img):
     mask = segmodel.predict(img)[0]
     return mask
 
-def display_mask(img, mask):
-    # Convert the grayscale mask to RGB
-    mask_rgb = np.stack([np.zeros_like(mask), mask, np.zeros_like(mask)], axis=-1)
-    # Convert the image to PIL format
-    img_pil = Image.fromarray(img)
-    # Convert the mask to PIL format
-    mask_pil = Image.fromarray((mask_rgb * 255).astype(np.uint8))
-    # Overlay the mask on the image
-    overlaid_img = Image.blend(img_pil, mask_pil, alpha=0.5)
-    return overlaid_img
-
-def redact_image(img):
-    img = img.resize((256, 256))
-    mask = get_seg_mask(img)
-    car_mask = ban_cars_mask[mask]
-    return display_mask(img, car_mask)
-
-def flip(img):
-    return np.flipud(img)
-
 
-def app(img):
+def app(img, prompt):
     start_time = datetime.now().timestamp()
     img = img.resize(working_size)
     mask = ban_cars_mask[get_seg_mask(img)]
     overlay_img = Image.fromarray(np.stack([img[:, :, 0], mask / 2, img[:,:,2]], axis=-1))
     end_time = datetime.now().timestamp()
     draw = ImageDraw.Draw(overlay_img)
-    draw.text((10, 10), f"Duration: {int(1000 * (end_time - start_time))}ms", fill=(255, 255, 255))
+    draw.text((10, 10), f"Duration: {int(1000 * (end_time - start_time))}ms\n<{prompt}>", fill=(255, 255, 255))
     return overlay_img
 
 #ideally:
 #iface = gr.Interface(app, gr.Image(sources=["webcam"], streaming=True), "image", live=True)
-iface = gr.Interface(app, gr.Image(), "image")
+iface = gr.Interface(app, [gr.Image(), gr.Textbox(value=default_inpainting_prompt)], "image")
 iface.launch()