Update app.py

app.py

@@ -18,12 +18,13 @@ import os
 import cv2
 import torch
 import glob
+from PIL import Image
 
 # Model
 
 def predict(text,img):
   model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
-  img1 = Image.open(img)
+  img1 = Image.open(img).convert("RGB")
   results = model(img1)  
   dirpath = tempfile.mkdtemp()
   results.crop(save_dir=dirpath)