edugp commited on
Commit
7b3d1d9
1 Parent(s): b9a0770

Support png and jpg

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def load_tokenizer_and_model():
36
  tokenizer, model = load_tokenizer_and_model()
37
 
38
  st.title("Caption Scoring")
39
- uploaded_file = st.file_uploader("Choose an image...", type="jpg")
40
  text_input = st.text_input("Type a caption")
41
 
42
  if uploaded_file is not None and text_input:
36
  tokenizer, model = load_tokenizer_and_model()
37
 
38
  st.title("Caption Scoring")
39
+ uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg"])
40
  text_input = st.text_input("Type a caption")
41
 
42
  if uploaded_file is not None and text_input: