fffiloni commited on
Commit
dbfb439
1 Parent(s): 6e5a5ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def train(prompt, init_image, trn_text, trn_steps):
38
  with torch.no_grad():
39
  torch.cuda.empty_cache()
40
 
41
- return "Training is finished !", gr.Update(value=0), gr.Update(value=0)
42
 
43
  def generate(prompt, init_image, trn_text, trn_steps):
44
  init_image = Image.open(init_image).convert("RGB")
 
38
  with torch.no_grad():
39
  torch.cuda.empty_cache()
40
 
41
+ return "Training is finished !", gr.update(value=0), gr.update(value=0)
42
 
43
  def generate(prompt, init_image, trn_text, trn_steps):
44
  init_image = Image.open(init_image).convert("RGB")