Read model stream from start

protectai · Mar 28, 2024 · 6e1ea14 · 6e1ea14
1 parent f49d8ac
commit 6e1ea14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modelscan/model.py b/modelscan/model.py
@@ -51,4 +51,4 @@ def get_stream(self) -> IO[bytes]:
         if not self._stream:
             raise ModelDataEmpty("Model data is empty.")
 
-        return self._stream
+        return self._stream.seek(0)