From 0781c2152dad8ebc1014f53df0362594e266f599 Mon Sep 17 00:00:00 2001
From: Baiju Meswani <bmeswani@microsoft.com>
Date: Thu, 9 Jan 2025 21:39:11 +0000
Subject: [PATCH] Fix compiler error

---
 src/python/python.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/python/python.cpp b/src/python/python.cpp
index 52a0870e8..9f6926eb4 100644
--- a/src/python/python.cpp
+++ b/src/python/python.cpp
@@ -495,7 +495,7 @@ PYBIND11_MODULE(onnxruntime_genai, m) {
         return std::make_shared<Images>(std::move(image_raw_data), image_datas.size());
       });
 
-  pybind11::class_<Audios, std::shared_ptr<Images>>(m, "Audios")
+  pybind11::class_<Audios, std::shared_ptr<Audios>>(m, "Audios")
       .def_static("open", [](pybind11::args audio_paths) {
         if (audio_paths.empty())
           throw std::runtime_error("No audios provided");
@@ -510,7 +510,7 @@ PYBIND11_MODULE(onnxruntime_genai, m) {
           audio_paths_vector.push_back(audio_paths_string.back().c_str());
         }
 
-        return std::make_shared<Audios>(LoadAudios(audio_paths_vector));
+        return std::shared_ptr<Audios>(LoadAudios(audio_paths_vector));
       });
 
   pybind11::class_<PyNamedTensors, std::shared_ptr<PyNamedTensors>>(m, "NamedTensors");