From 46b613fef20e33719666516a3fa4f27d1fd559d4 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Mon, 24 May 2021 19:37:04 -0400 Subject: [PATCH] uploading image template processed faces works! --- src/job_queries/w2l_download_job_queries.rs | 4 +++- src/script_execution/wav2lip_process_upload_command.rs | 2 +- src/w2l_download_job.rs | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/job_queries/w2l_download_job_queries.rs b/src/job_queries/w2l_download_job_queries.rs index 0141f95b3d..16d27409ea 100644 --- a/src/job_queries/w2l_download_job_queries.rs +++ b/src/job_queries/w2l_download_job_queries.rs @@ -135,7 +135,9 @@ SET original_download_url = ?, private_bucket_hash = ?, private_bucket_object_name = ?, - private_bucket_cached_faces_object_name = ? + private_bucket_cached_faces_object_name = ?, + public_bucket_hash = 'TODO', + public_bucket_object_name = 'TODO' "#, model_token, template_type, diff --git a/src/script_execution/wav2lip_process_upload_command.rs b/src/script_execution/wav2lip_process_upload_command.rs index d5eec7eba7..ae2268b5dc 100644 --- a/src/script_execution/wav2lip_process_upload_command.rs +++ b/src/script_execution/wav2lip_process_upload_command.rs @@ -46,7 +46,7 @@ impl Wav2LipPreprocessClient { command.push_str(&self.checkpoint_path); command.push_str(" --image_or_video_filename "); command.push_str(image_or_video_filename); - command.push_str(" --output_cached_faces_filename"); + command.push_str(" --output_cached_faces_filename "); command.push_str(output_cached_faces_filename); if is_image { diff --git a/src/w2l_download_job.rs b/src/w2l_download_job.rs index a7f06e0ee2..0efc0c4af4 100644 --- a/src/w2l_download_job.rs +++ b/src/w2l_download_job.rs @@ -241,7 +241,7 @@ async fn process_job(downloader: &Downloader, job: &W2lTemplateUploadJobRecord) let cached_faces_filename = format!("{}_detected_faces.pickle", &download_filename); let is_image = false; // TODO: Don't always treat as video. - let spawn_process = true; + let spawn_process = false; downloader.w2l_processor.execute( &download_filename,