diff --git a/src/main.py b/src/main.py index d49473c..b0dcc4d 100644 --- 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