Merge pull request #58 from jainkunal/patch-1

fix: Session is not initialized.
gizatechxyz · Jun 10, 2024 · 9b43136 · 9b43136
2 parents 8e5fb6a + e42bbf7
commit 9b43136
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/giza/agents/model.py b/giza/agents/model.py
@@ -93,14 +93,14 @@ def __init__(
             self.uri = self._retrieve_uri()
             self.endpoint_id = self._get_endpoint_id()
             self._cache = Cache(os.path.join(os.getcwd(), "tmp", "cachedir"))
-            self.session = self._set_session()
             if output_path is not None:
                 self._output_path = output_path
             else:
                 self._output_path = os.path.join(
                     tempfile.gettempdir(),
                     f"{self.model_id}_{self.version_id}_{self.model.name}",
                 )
+            self.session = self._set_session()
             self._download_model()
 
     def _get_endpoint_id(self) -> int: