boazchung commited on
Commit
01dadd8
โ€ข
1 Parent(s): 8ee0687

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,9 +1,8 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
  from PIL import Image
4
-
5
- pipeline = pipeline(task="image-classification",
6
- model="julien-c/hotdog-not-hotdog")
7
 
8
  st.title("Hot Dog? Or Not?")
9
 
@@ -14,7 +13,8 @@ if file_name is not None:
14
 
15
  image = Image.open(file_name)
16
  col1.image(image, use_column_width=True)
17
- predictions = pipeline(image)
 
18
 
19
  col2.header("Probabilities")
20
  for p in predictions:
 
1
  import streamlit as st
2
  from transformers import pipeline
3
  from PIL import Image
4
+ # TO-DD: ??? ๋ถ€๋ถ„์˜ ์ฝ”๋“œ๋ฅผ ์™„์„ฑํ•˜์‹œ์˜ค
5
+ pipeline = pipeline(???)
 
6
 
7
  st.title("Hot Dog? Or Not?")
8
 
 
13
 
14
  image = Image.open(file_name)
15
  col1.image(image, use_column_width=True)
16
+ # TO-DO: ??? ๋ถ€๋ถ„์˜ ์ฝ”๋“œ๋ฅผ ์™„์„ฑํ•˜์‹œ์˜ค
17
+ predictions = ???
18
 
19
  col2.header("Probabilities")
20
  for p in predictions: