adding app with CLIP image segmentation

app.py

@@ -37,7 +37,7 @@ def detect_using_clip(image,prompts=[],threshould=0.4):
 
     return predicted_masks
 
-def visualize_images(image,predicted_images,brightness=15):
+def visualize_images(image,predicted_images,brightness=15,contrast=1.8):
     alpha = 0.7
     image_resize = cv2.resize(image,(352,352))
     resize_image_copy = image_resize.copy()
@@ -45,9 +45,9 @@ def visualize_images(image,predicted_images,brightness=15):
     for mask_image in predicted_images:
         resize_image_copy = cv2.addWeighted(resize_image_copy,alpha,mask_image,1-alpha,10)
 
-    return cv2.convertScaleAbs(resize_image_copy, alpha=1.8, beta=brightness)     
+    return cv2.convertScaleAbs(resize_image_copy, alpha=contrast, beta=brightness)     
 
-def shot(brightness,image, labels_text):
+def shot(brightness,contrast,image,labels_text):
     if "," in labels_text:
         prompts = labels_text.split(',')
     else:
@@ -55,20 +55,23 @@ def shot(brightness,image, labels_text):
     prompts = list(map(lambda x: x.strip(),prompts))
     predicted_images  = detect_using_clip(image,prompts=prompts)
 
-    category_image = visualize_images(image=image,predicted_images=predicted_images,brightness=brightness)
+    category_image = visualize_images(image=image,predicted_images=predicted_images,brightness=brightness,contrast=contrast)
     return category_image
 
 iface = gr.Interface(fn=shot,
-                    inputs = [gr.Slider(5, 50, value=15, label="Brightness", info="Choose between 5 and 50"),"image","text"],
+                    inputs = [
+                        gr.Slider(5, 50, value=15, label="Brightness", info="Choose between 5 and 50"),
+                        gr.Slider(1, 5, value=1.5, label="Contrast", info="Choose between 1 and 5"),
+                        "image",
+                        "text"
+                        ],
                     outputs = "image",
                     description ="Add an Image and lists of category to be detected separated by commas(atleast 2 )",
                     title = "Zero-shot Image Segmentation with Prompt ",
                     examples=[
-                        [15,"images/room.jpg","bed, table, plant, light, window,light"],
-                        [10,"images/image2.png","banner, building,door, sign,"],
                         [19,"images/seats.jpg","door,table,chairs"],
-                        [20,"images/vegetables.jpg","carrot,radish,beans,potato,brnjal,basket"],
-                        [17,"images/room2.jpg","door,platns,dog,coffe table,mug,pillow,table lamp,carpet,pictures,door,clock"]
+                        [20,"images/vegetables.jpg","carrot,white radish,brinjal,basket,potato"],
+                        [17,"images/room2.jpg","door, plants, dog, coffe table, table lamp, carpet, door"]
                         ],
                     # allow_flagging=False, 
                     # analytics_enabled=False,