Update app.py

app.py

@@ -49,7 +49,7 @@ def visualize_images(image,predicted_images,brightness=15,contrast=1.8):
 
     return cv2.convertScaleAbs(resize_image_copy, alpha=contrast, beta=brightness)     
 
-def shot(brightness,contrast,image,labels_text):
+def shot(alpha,beta,image,labels_text):
     if "," in labels_text:
         prompts = labels_text.split(',')
     else:
@@ -63,7 +63,7 @@ def shot(brightness,contrast,image,labels_text):
     resize_image = cv2.resize(image,(352,352))
 
     predicted_images  = detect_using_clip(image,prompts=prompts)
-    category_image = overlay_masks(resize_image,np.stack(predicted_images,-1),labels=mask_labels,colors=cmap,alpha=0.4,beta=1)
+    category_image = overlay_masks(resize_image,np.stack(predicted_images,-1),labels=mask_labels,colors=cmap,alpha=alpha,beta=beta)
 
     return category_image