Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import cv2
 from PIL import Image
 from ultralytics import YOLO
+import torch
 
 # Define available YOLO models
 available_models = {
@@ -12,6 +13,8 @@ available_models = {
     # Add more models as needed
 }
 
+
+
 # Create a function to perform image segmentation using the selected model
 def segment_image(input_image, selected_model):
     # Resize the input image to 255x255
@@ -33,7 +36,7 @@ def segment_image(input_image, selected_model):
     overlay_image = img.copy()
 
     # Apply the resized mask to the overlay image
-    overlay_image[resized_mask > 0] = [100, 0, 0]  # Overlay in green
+    overlay_image[resized_mask > 0] = [50, 0, 0]  # Overlay in green
 
     # Convert the overlay image to PIL format
     overlay_pil = Image.fromarray(overlay_image)