Spaces:
Sleeping
Sleeping
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")
|