Update app.py

app.py

@@ -42,6 +42,7 @@ import mediapipe as mp
 from transformers import pipeline
 from skimage.measure import label, regionprops
 import matplotlib.pyplot as plt
+import spaces
 
 
 def _normalized_to_pixel_coordinates(
@@ -101,7 +102,7 @@ def create_bounding_box_mask(image):
     return bounding_mask
 
 
-
+@spaces.GPU
 def segment_2(image_np, coordinates):
     OVERLAY_COLOR = (255, 105, 180)  # Rose
 
@@ -164,7 +165,7 @@ def segment_2(image_np, coordinates):
 
         return output_image,bbox_mask_image
     
-
+@spaces.GPU
 def generate_2(image_file_path, bbox_image, prompt):
 
     # Read image