Update app.py

app.py

@@ -3,7 +3,6 @@ from model import ENet
 import numpy as np
 import torch
 import os
-import cv2
 
 num_classes = 19
 model = ENet(num_classes)
@@ -12,8 +11,8 @@ model_path = "path"
 model.load_state_dict(model_path)
 
 raw_image = st.file_uploader('Raw Input Image')
-if raw_image is not None: 
-  image = cv2.imread(raw_image) 
+if raw_image is not None:
+  image = np.asarray(raw_image)
   output = model(image)
   h, w = output.shape