Update app.py

app.py

@@ -38,7 +38,7 @@ def train(prompt, init_image, trn_text, trn_steps):
     with torch.no_grad():
         torch.cuda.empty_cache()
     
-    return "Training is finished !", gr.Update(value=0), gr.Update(value=0)
+    return "Training is finished !", gr.update(value=0), gr.update(value=0)
    
 def generate(prompt, init_image, trn_text, trn_steps):
     init_image = Image.open(init_image).convert("RGB")