update

gradio_app.py

@@ -9,7 +9,7 @@ import uuid
 from drag_gan import drag_gan, stylegan2
 from stylegan2.inversion import inverse_image
 
-device = 'cpu'
+device = 'cuda'
 
 
 SIZE_TO_CLICK_SIZE = {
@@ -337,7 +337,7 @@ def main():
 if __name__ == '__main__':
     import argparse
     parser = argparse.ArgumentParser()
-    parser.add_argument('--device', default='cuda')
+    parser.add_argument('--device', default='cpu')
     parser.add_argument('--share', action='store_true')
     parser.add_argument('-p', '--port', default=None)
     parser.add_argument('--ip', default=None)