vancauwe commited on
Commit
140650b
·
1 Parent(s): a50f169

fix: pillow from numpy array

Browse files
Files changed (1) hide show
  1. src/main.py +1 -1
src/main.py CHANGED
@@ -333,7 +333,7 @@ def main() -> None:
333
  # display the image (use cached version, no need to reread)
334
  col1.image(st.session_state.image, use_column_width=True)
335
  # and then run inference on the image
336
- hotdog_image = Image.open(st.session_state.image)
337
  predictions = pipeline_hot_dog(hotdog_image)
338
 
339
  col2.header("Probabilities")
 
333
  # display the image (use cached version, no need to reread)
334
  col1.image(st.session_state.image, use_column_width=True)
335
  # and then run inference on the image
336
+ hotdog_image = Image.fromarray(st.session_state.image)
337
  predictions = pipeline_hot_dog(hotdog_image)
338
 
339
  col2.header("Probabilities")