update app.py

app.py

@@ -68,11 +68,13 @@ def visualize_images(image,predicted_images,brightness=15,contrast=1.8):
 
 def shot(alpha,beta,image,labels_text):
     print(labels_text)
+    
     if "," in labels_text:
         prompts = labels_text.split(',')
     else:
         prompts = [labels_text]
     print(prompts)
+    
     prompts = list(map(lambda x: x.strip(),prompts))
 
     mask_labels = [f"{prompt}_{i}" for i,prompt in enumerate(prompts)]
@@ -81,7 +83,7 @@ def shot(alpha,beta,image,labels_text):
 
     predicted_masks  = detect_using_clip(image,prompts=prompts)
     bool_masks = [predicted_mask.astype('bool') for predicted_mask in predicted_masks]
-    category_image = overlay_masks(resize_image,np.stack(bool_masks,-1),labels=mask_labels,colors=cmap,alpha=alpha,beta=beta)
+    category_image = overlay_masks(image,np.stack(bool_masks,-1),labels=mask_labels,colors=cmap,alpha=alpha,beta=beta)
 
     return category_image