Update app.py

app.py

@@ -297,8 +297,8 @@ def train(*inputs):
     torch.cuda.empty_cache()
     if(which_model == "v1-5"):
         print("Adding Safety Checker to the model...")
-        shutil.copytree(f"{safety_checker}/feature_extractor", "output_model/feature_extractor")
-        shutil.copytree(f"{safety_checker}/safety_checker", "output_model/safety_checker")
+        shutil.copytree(f"{safety_checker}/feature_extractor", "output_model/feature_extractor", dirs_exist_ok=True)
+        shutil.copytree(f"{safety_checker}/safety_checker", "output_model/safety_checker", dirs_exist_ok=True)
         shutil.copy(f"model_index.json", "output_model/model_index.json")
     
     if(not remove_attribution_after):