clean execute_task

app.py

@@ -118,8 +118,9 @@ def generate_image(setup_args, num_iterations):
         def run_main():
             thread_status["running"] = True  # Mark thread as running
             try:
-                result_container["best_image"], result_container["total_init_rewards"], result_container["total_best_rewards"] = execute_task(
-                    args, trainer, device, dtype, shape, enable_grad, settings, progress_callback)
+                execute_task(
+                    args, trainer, device, dtype, shape, enable_grad, settings, progress_callback
+                )
             except torch.cuda.OutOfMemoryError as e:
                 print(f"CUDA Out of Memory Error: {e}")
                 error_status["error_occurred"] = True