Fix bug

app.py

@@ -32,7 +32,7 @@ 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.name)
+    pred_dict = predict(model, image_file)
     print('Fixing the preds')
     boxes, image = prepare_prediction(pred_dict)
     print('Plotting')

model.py

@@ -36,7 +36,8 @@ def get_checkpoint(checkpoint_path):
     return fixed_state_dict
 
 def predict(model, image):
-    img = PIL.Image.open(image)
+    #img = PIL.Image.open(image)
+    img = PIL.Image(image.read())
     print(img.shape)
     class_map = ClassMap(classes=['Waste'])
     transforms = tfms.A.Adapter([