Update src/streamlit_app.py

src/streamlit_app.py

@@ -105,8 +105,7 @@ st.write("Detect whether a person is wearing a face mask or not.")
 option = st.radio("Select Input Method", ["Upload Image", "Image URL"])
 
 if option == "Upload Image":
-    uploaded_file = st.file_uploader("Please upload an image.")
-
+    uploaded_file = st.file_uploader("Please upload an image.", type=["jpg", "jpeg", "png"])
     if uploaded_file is not None:
         img = PILImage.create(uploaded_file)
         detect_modify(img, model, device, conf=conf, imgsz=imgsz, conf_thres=conf_thres, iou_thres=iou_thres)