Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import torch
 from PIL import Image, ImageFilter, ImageDraw
 from ultralytics import YOLO
+import spaces
 
 MODEL_PATH = "./models/face_yolov8n_v2.pt"
 MODEL = YOLO(MODEL_PATH)
@@ -37,7 +38,8 @@ def create_rounded_rectangle_mask(image:Image.Image, radius, alpha=255):
 
 def max_rounding_radius(rect_coords):
     # Extract the coordinates of the rectangle [x0, y0, x1, y1]
-    rect_coords = [coord-1 for coord in rect_coords]
+    rect_coords = [coord
+-1 for coord in rect_coords]
     x0, y0, x1, y1 = rect_coords
 
     # Calculate the width and height of the rectangle
@@ -50,11 +52,12 @@ def max_rounding_radius(rect_coords):
     # Calculate the maximum radius as half of the smallest dimension
     return min_dimension // 2
 
+@spaces.GPU(enable_queue=True)
 def generate_image(source_image:Image.Image, confidence=0.3, radius=50, blur_amount=10, margin=0):
     if source_image is None:
         return source_image
 
-    device = "cuda" if torch.cuda.is_available() else "cpu"
+    device = "cuda"# if torch.cuda.is_available() else "cpu"
     pred = MODEL(source_image, conf=confidence, device=device)
     bboxes = pred[0].boxes.xyxy.cpu().numpy()
     if bboxes.size == 0: