Update app.py

app.py

@@ -33,6 +33,7 @@ def process_images(files):
     # return 'Before: {}\nAfter: {}'.format(files_str, saved_files_str), saved_files[0]
 
     command = [
+        "CUDA_VISIBLE_DEVICES=0",
         "python",
         "-W", "ignore",
         "./coarse_init_infer.py",