Update tasks.py

tasks.py

@@ -108,7 +108,7 @@ def optimize_on_server(content, style, result_image_placeholder):
     style = pil_resize_long_edge_to(style, 1024)
     style.save(style_path)
     files = {'style-image': open(style_path, "rb"), "content-image": open(content_path, "rb")}
-    print("start-optimizing")
+    print("start-optimizing. Time: ", datetime.datetime.now())
     url = WORKER_URL + "/upload"
     task_id_res = requests.post(url, files=files)
     if task_id_res.status_code != 200:
@@ -140,4 +140,4 @@ def optimize_params(effect, preset, content, style, result_image_placeholder):
                                         write_video=False, base_dir=base_dir,
                                         iter_callback=lambda i: progress_bar.progress(
                                             float(i) / ST_CONFIG["n_iterations"]))
-    st.session_state["effect_input"], st.session_state["result_vp"]  = content_img_cuda.detach(), vp.cuda().detach()
+    st.session_state["effect_input"], st.session_state["result_vp"]  = content_img_cuda.detach(), vp.cuda().detach()