Upload app.py

app.py

@@ -559,21 +559,10 @@ except:
 
 
 
-if device == "cuda":
-
-    PYTORCH_CUDA_ALLOC_CONF = {
-        "max_split_size_mb": 8000
-    }
-    torch.cuda.max_memory_allocated(
-        device = device
-    )
-    torch.cuda.empty_cache()
-
-
-
 if script_being_run_on_hugging_face == 1:
 
     use_custom_hugging_face_cache_dir = 0
+    auto_save_imagery = 0
     show_messages_in_modal_on_page = 0
 
     ending_html = """
@@ -583,6 +572,18 @@ If you would like to download this app to run offline on a Windows computer that
 
 
 
+if device == "cuda":
+
+    PYTORCH_CUDA_ALLOC_CONF = {
+        "max_split_size_mb": 8000
+    }
+    torch.cuda.max_memory_allocated(
+        device = device
+    )
+    torch.cuda.empty_cache()
+
+
+
 saved_images_dir = main_dir + "/" + saved_images_folder_name
 
 hugging_face_cache_dir = main_dir + "/" + cache_directory_folder_name