Update app.py

app.py

@@ -99,7 +99,7 @@ def build():
 			out5 = gr.Image()
 			out6 = gr.Image()
 		with gr.Row():
-			btn3 = gr.Button("Download")
+			# btn3 = gr.Button("Download")
 			caixa = gr.File(file_count="multiple", file_types=["text", ".json", ".csv", "image"])
 
 		with gr.Row(visible=False):
@@ -122,11 +122,11 @@ def build():
 		b5=out3.change(im_fn,[prompt,noise_level,h_variavel],[out4,], show_progress=True)
 		b6=out4.change(im_fn,[prompt,noise_level,h_variavel],[out5,], show_progress=True)
 		b7=out5.change(im_fn,[prompt,noise_level,h_variavel],[out6,], show_progress=True)
-		b8=out6.change(noth,None,[message], show_progress=False)
+		b8=out6.change(noth,None,[message], show_progress=False).then(zip_files,None,[caixa])
 		swi=t_switch.change(clear,None,[t_switch,fac_b], cancels=[sta,b2,b3,b4,b5,b6,b7],show_progress=False)
 		#btn2.click(noth,None,message,cancels=[b1,sta,b2,b3,b4,b5,swi],show_progress=False)
 		btn2.click(clear_all, None,[fac_b,prompt,out1,out2,out3,out4,out5,out6,t_state,t_switch,message],cancels=[b1,sta,b2,b3,b4,b5,b6,b7,b8,swi],show_progress=False)
-		btn3.click(zip_files,None,[caixa],show_progress=False)
+		# btn3.click(zip_files,None,[caixa],show_progress=False)
 		# caixa.change(noth,None,[message],show_progress=False)
 	b.queue(concurrency_count=100).launch(show_api=False)
 build()