Update app.py

app.py

@@ -352,7 +352,7 @@ def make_dataloaders2(batch_size=16, n_workers=4, pin_memory=True, **kwargs): #
                             pin_memory=pin_memory)
     return dataloader
 
-file_up = st.file_uploader("Upload an jpg image", type="jpg")
+file_up = st.file_uploader("Upload an jpg image", type=['png', 'jpeg', 'jpg'])
 if file_up is not None: 
     im = Image.open(file_up)
     st.text(body=f"Size of uploaded image {im.shape}")