mod cuda option

pipeline.py

@@ -17,7 +17,7 @@ from diffusers import AutoPipelineForInpainting
 
 class SmileGen:
     def __init__(self, device='cuda'):
-        self.device = device
+        self.device = device if torch.cuda.is_available() else 'cpu'
     
     def face_detection(self, image):
         face_det = YOLO(hf_hub_download(repo_id="arnabdhar/YOLOv8-Face-Detection", filename="model.pt")).to(self.device)