Update app.py
Browse files
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 !"
|
42 |
|
43 |
def generate(prompt, init_image, trn_text, trn_steps):
|
44 |
init_image = Image.open(init_image).convert("RGB")
|
@@ -151,7 +151,7 @@ with gr.Blocks(css=css) as block:
|
|
151 |
|
152 |
gr.HTML(article)
|
153 |
|
154 |
-
train_btn.click(fn=train, inputs=[prompt_input,image_init,trn_text,trn_steps], outputs=[training_status])
|
155 |
gen_btn.click(fn=generate, inputs=[prompt_input,image_init,trn_text,trn_steps], outputs=[image_output])
|
156 |
|
157 |
block.queue(max_size=12).launch(show_api=False)
|
|
|
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")
|
|
|
151 |
|
152 |
gr.HTML(article)
|
153 |
|
154 |
+
train_btn.click(fn=train, inputs=[prompt_input,image_init,trn_text,trn_steps], outputs=[training_status, trn_text, trn_steps])
|
155 |
gen_btn.click(fn=generate, inputs=[prompt_input,image_init,trn_text,trn_steps], outputs=[image_output])
|
156 |
|
157 |
block.queue(max_size=12).launch(show_api=False)
|