Update app.py

app.py

@@ -1,6 +1,3 @@
-import os
-os.system('pip install gradio==4.29.0')
-
 import random
 from dataclasses import dataclass
 from typing import Any, List, Dict, Optional, Union, Tuple
@@ -183,7 +180,7 @@ def draw_classification_boxes(image_with_insects: np.ndarray, detections: List[D
     return image_with_insects
 
 def process_image(image):
-    labels = ["ant", "beetle", "butterfly", "caterpillar", "dragonfly"]
+    labels = ["insect"]
     original_image, detections = grounded_segmentation(image, labels, threshold=0.3, polygon_refinement=True)
     masked_image = plot_detections(original_image, detections)
     insect_masks = extract_insect_masks(original_image, detections)
@@ -196,5 +193,5 @@ gr.Interface(
     fn=process_image,
     inputs=gr.Image(type="pil"),
     outputs=[gr.Image(type="numpy"), gr.Image(type="numpy"), gr.Image(type="numpy")],
-    title="Insect Detection and Masking"
+    title="🐞 Insect Detection and Masking"
 ).launch()