Fixed upload file bug

app.py

@@ -32,7 +32,8 @@ image_file = st.file_uploader("Upload Images", type=["png","jpg","jpeg"])
 if image_file is not None:
     print(image_file)
     print('Getting predictions')
-    pred_dict = predict(model, image_file)
+    data = image_file.read()
+    pred_dict = predict(model, data)
     print('Fixing the preds')
     boxes, image = prepare_prediction(pred_dict)
     print('Plotting')

model.py

@@ -1,3 +1,4 @@
+from io import BytesIO
 from icevision import *
 import collections
 import PIL
@@ -37,7 +38,7 @@ def get_checkpoint(checkpoint_path):
 
 def predict(model, image):
     #img = PIL.Image.open(image)
-    img = PIL.Image(image.read())
+    img = PIL.Image.open(BytesIO(image))
     print(img.shape)
     class_map = ClassMap(classes=['Waste'])
     transforms = tfms.A.Adapter([