Update main.cpp

Signed-off-by: quic-khrahul <131336334+quic-khrahul@users.noreply.github.com>

Author: quic-khrahul

RB5/linux_kernel_5_x/AI-ML-apps/AI_Age_Gender_Emotion_Solution/src/main.cpp

@@ -110,19 +110,18 @@ void InferenceThread(void *sol_conf,int model_id, int size)
     uint32_t frames = 0;
     do
     {  
+	unique_lock<mutex> lock(mu);
+	cond.wait(lock, [&](){
+		return model_id == current_model;
+	}
+        );
         if(current_model == face_detect)
         {
             ret = gDecodeQueue->Dequeue(item, 300);
         }
         if (ret == 0)
         {
             frames += 1;
-            
-            unique_lock<mutex> lock(mu);
-            cond.wait(lock, [&](){
-                return model_id == current_model;
-                }
-            );
             auto start1 = chrono::steady_clock::now();
             shInference->Inference(item);
             auto end1 = chrono::steady_clock::now();