MinxuanQin commited on
Commit
5cca687
1 Parent(s): 6f67cca

add more pic format

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ model = ViltForQuestionAnswering.from_pretrained("Minqin/carets_vqa_finetuned")
13
 
14
  orig_model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
15
 
16
- uploaded_file = st.file_uploader("Please upload one image (jpg)", type="jpg")
17
 
18
  question = st.text_input("Type here one question on the image")
19
  if uploaded_file is not None:
 
13
 
14
  orig_model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
15
 
16
+ uploaded_file = st.file_uploader("Please upload one image", type=["jpg", "png", "bmp", "jpeg"])
17
 
18
  question = st.text_input("Type here one question on the image")
19
  if uploaded_file is not None: