Fix bug

app.py

@@ -22,7 +22,7 @@ def plot_img_no_mask(image, boxes):
         pt1 = (x1, y1)
         pt2 = (x2, y2)
         cv2.rectangle(image, pt1, pt2, (220,0,0), thickness=5)
-        
+
     plt.axis('off')
     ax.imshow(image)
     fig.savefig("img.png", bbox_inches='tight')
@@ -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)
+    pred_dict = predict(model, image_file.name)
     print('Fixing the preds')
     boxes, image = prepare_prediction(pred_dict)
     print('Plotting')