Update app.py

app.py

@@ -162,7 +162,7 @@ def train(*inputs):
     convert("output_model", "model.ckpt")
     shutil.rmtree('instance_images')
     shutil.make_archive("diffusers_model", 'zip', "output_model")
-    shutil.rmtree("output_model")
+    #shutil.rmtree("output_model")
     torch.cuda.empty_cache()
     return [gr.update(visible=True, value=["diffusers_model.zip", "model.ckpt"]), gr.update(visible=True), gr.update(visible=True)]