diff --git a/crates/_types/enums/src/by_table/generic_inference_jobs/inference_category.rs b/crates/_types/enums/src/by_table/generic_inference_jobs/inference_category.rs index 03147bc224..c776b40dd5 100644 --- a/crates/_types/enums/src/by_table/generic_inference_jobs/inference_category.rs +++ b/crates/_types/enums/src/by_table/generic_inference_jobs/inference_category.rs @@ -55,6 +55,7 @@ impl InferenceCategory { "text_to_speech" => Ok(Self::TextToSpeech), "voice_conversion" => Ok(Self::VoiceConversion), "video_filter" => Ok(Self::VideoFilter), + "mocap" => Ok(Self::Mocap), _ => Err(format!("invalid value: {:?}", value)), } } @@ -67,6 +68,7 @@ impl InferenceCategory { Self::TextToSpeech, Self::VoiceConversion, Self::VideoFilter, + Self::Mocap, ]) } } diff --git a/crates/service/job/inference_job/src/job/job_types/mocap/mocap_net/mocapnet_inference_command.rs b/crates/service/job/inference_job/src/job/job_types/mocap/mocap_net/mocapnet_inference_command.rs index c040c86042..e908dce49a 100644 --- a/crates/service/job/inference_job/src/job/job_types/mocap/mocap_net/mocapnet_inference_command.rs +++ b/crates/service/job/inference_job/src/job/job_types/mocap/mocap_net/mocapnet_inference_command.rs @@ -33,7 +33,7 @@ static IGNORED_ENVIRONMENT_VARS : Lazy> = Lazy::new(|| { #[derive(Clone)] pub struct MocapnetInferenceCommand { /// Where the code lives - mocapnet_root_code_directory: PathBuf, + pub(crate) mocapnet_root_code_directory: PathBuf, /// A single executable script or a much larger bash command. executable_or_command: ExecutableOrCommand, @@ -152,8 +152,7 @@ impl MocapnetInferenceCommand { // get video path let video_path = args.video_file.as_ref(); command.push_str(&format!(" --from {}", path_to_string(video_path))); - - command.push_str(" --all --save "); + command.push_str(" --all --save --headless 2"); // ===== End Python Args ===== diff --git a/crates/service/job/inference_job/src/job/job_types/mocap/mocap_net/process_job.rs b/crates/service/job/inference_job/src/job/job_types/mocap/mocap_net/process_job.rs index 206e4073c2..8b49906ecd 100644 --- a/crates/service/job/inference_job/src/job/job_types/mocap/mocap_net/process_job.rs +++ b/crates/service/job/inference_job/src/job/job_types/mocap/mocap_net/process_job.rs @@ -15,7 +15,7 @@ use mimetypes::mimetype_for_file::get_mimetype_for_file; use mysql_queries::payloads::generic_inference_args::generic_inference_args::PolymorphicInferenceArgs::Mc; use mysql_queries::queries::generic_inference::job::list_available_generic_inference_jobs::AvailableInferenceJob; use mysql_queries::queries::media_files::create::insert_media_file_from_mocapnet::{insert_media_file_from_mocapnet, InsertArgs}; -use crate::job::job_types::mocap::mocap_net::mocapnet_inference_command::InferenceArgs; +use crate::job::job_types::mocap::mocap_net::mocapnet_inference_command::{InferenceArgs, MocapnetInferenceCommand}; use crate::job::job_loop::job_success_result::{JobSuccessResult, ResultEntity}; use crate::job::job_loop::process_single_job_error::ProcessSingleJobError; @@ -95,11 +95,12 @@ pub async fn process_job(args: MocapNetProcessJobArgs<'_>) -> Result) -> Result) -> Result) -> Result) -> Result (app: App, server_environment: ServerEnvironment) -> // ==================== Mocap ======================== let mut app = RouteBuilder::from_app(app) - .add_get("/v1/mocap/mocapnet/create", enqueue_mocapnet_handler) + .add_post("/v1/mocap/mocapnet/create", enqueue_mocapnet_handler) .into_app(); // ==================== "Generic" Inference ====================