Update app.py

app.py

@@ -57,7 +57,7 @@ def generate_image(source_image:Image.Image, confidence=0.3, radius=50, blur_amo
     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: