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
|