vancauwe commited on
Commit
4f11b2f
·
1 Parent(s): 7c16c6e

fix: change pipeline name

Browse files
Files changed (1) hide show
  1. 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
- pipeline = 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,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 = pipeline(st.session_state.image)
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