Spaces:
Running
Running
fix: pillow from numpy array
Browse files- 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.
|
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")
|