Update face_enhancer.py

face_enhancer.py

@@ -40,7 +40,7 @@ def get_available_enhancer_names():
     return available
 
 
-def load_face_enhancer_model(name='GFPGAN', device="cuda"):
+def load_face_enhancer_model(name='GFPGAN', device='cuda' if torch.cuda.is_available() else 'cpu'):
     assert name in get_available_enhancer_names() + cv2_interpolations, f"Face enhancer {name} unavailable."
     if name in supported_enhancers.keys():
         model_path, model_runner = supported_enhancers.get(name)