Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import torch
 import cv2
+import numpy as np
 import torch.nn as nn
 from torchvision.ops import box_iou
 from PIL import Image
@@ -52,7 +53,7 @@ file_name = st.file_uploader("Upload oil palm tree image")
 if file_name is not None:
     col1, col2 = st.columns(2)
 
-    image = Image.open(file_name)
+    image = np.array(Image.open(file_name))
     col1.image(image, use_column_width=True)
     transformed = test_transforms(image= image)
     image_transformed = transformed["image"]