Update app.py

app.py

@@ -1,10 +1,8 @@
 import os 
 os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
-
-import gradio as gr
-import cv2
 os.system('pip install torch==1.9.0 torchvision==0.10.0')
 
+import gradio as gr
 # check pytorch installation: 
 import torch, torchvision
 print(torch.__version__, torch.cuda.is_available())
@@ -43,8 +41,7 @@ predictor = DefaultPredictor(cfg)
 
 
 def inference(img):
-    # im = cv2.imread(img.name)
-    im = cv2.imread(img)
+    im = np.asarray(Image.open(img).convert('RGB'))
     outputs = predictor(im)
 
     take = outputs['instances'].scores >= 0.5 #Threshold