Update app.py

app.py

@@ -156,8 +156,8 @@ class Demo:
         if self.training:
             return [gr.update(interactive=True, value='Train'), gr.update(value='Someone else is training... Try again soon'), None, gr.update()]
         # clear the diffusers
-        del self.diffuser
-        torch.cuda.empty_cache()
+        # del self.diffuser
+        # torch.cuda.empty_cache()
         
         if train_method == 'ESD-x':
 
@@ -190,7 +190,7 @@ class Demo:
 
         # del self.diffuser
         torch.cuda.empty_cache()
-        self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda').eval().half()
+        # self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda').eval().half()
         return [gr.update(interactive=True, value='Train'), gr.update(value='Done Training'), save_path, gr.Dropdown.update(choices=list(model_map.keys()), value='Custom')]