fix: pillow transform hotdog image

sdsc-ordes · Dec 21, 2024 · a50f169 · a50f169
1 parent 2761417
commit a50f169
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -333,7 +333,8 @@ def main() -> None:
             # display the image (use cached version, no need to reread)
             col1.image(st.session_state.image, use_column_width=True)
             # and then run inference on the image
-            predictions = pipeline_hot_dog(Image(st.session_state.image))
+            hotdog_image = Image.open(st.session_state.image)
+            predictions = pipeline_hot_dog(hotdog_image)
 
             col2.header("Probabilities")
             first = True