Spaces:
Sleeping
Sleeping
fix: change pipeline name
Browse files- src/main.py +2 -2
src/main.py
CHANGED
|
@@ -313,7 +313,7 @@ def main() -> None:
|
|
| 313 |
|
| 314 |
if tab_hotdogs.button("Get Hotdog Prediction"):
|
| 315 |
|
| 316 |
-
|
| 317 |
tab_hotdogs.title("Hot Dog? Or Not?")
|
| 318 |
|
| 319 |
if st.session_state.image is None:
|
|
@@ -326,7 +326,7 @@ def main() -> None:
|
|
| 326 |
# display the image (use cached version, no need to reread)
|
| 327 |
col1.image(st.session_state.image, use_column_width=True)
|
| 328 |
# and then run inference on the image
|
| 329 |
-
predictions =
|
| 330 |
|
| 331 |
col2.header("Probabilities")
|
| 332 |
first = True
|
|
|
|
| 313 |
|
| 314 |
if tab_hotdogs.button("Get Hotdog Prediction"):
|
| 315 |
|
| 316 |
+
pipeline_hot_dog = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
|
| 317 |
tab_hotdogs.title("Hot Dog? Or Not?")
|
| 318 |
|
| 319 |
if st.session_state.image is None:
|
|
|
|
| 326 |
# display the image (use cached version, no need to reread)
|
| 327 |
col1.image(st.session_state.image, use_column_width=True)
|
| 328 |
# and then run inference on the image
|
| 329 |
+
predictions = pipeline_hot_dog(st.session_state.image)
|
| 330 |
|
| 331 |
col2.header("Probabilities")
|
| 332 |
first = True
|