Spaces:
Sleeping
Sleeping
fix: pillow transform hotdog image
Browse files- src/main.py +2 -1
src/main.py
CHANGED
|
@@ -333,7 +333,8 @@ 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 |
-
|
|
|
|
| 337 |
|
| 338 |
col2.header("Probabilities")
|
| 339 |
first = True
|
|
|
|
| 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")
|
| 340 |
first = True
|