Update app.py

app.py

@@ -8,6 +8,8 @@ import shutil
 import gradio as gr
 from PIL import Image
 from runpy import run_path
+import numpy as np
+
 
 examples = [['sample1.png'],
             ['sample2.png']]
@@ -58,8 +60,9 @@ def inference(img, task, run_on):
         if torch.cuda.is_available():
             torch.cuda.ipc_collect()
             torch.cuda.empty_cache()
-        
-        img = cv2.cvtColor(cv2.imread(args.input_path), cv2.COLOR_BGR2RGB)
+
+        open_cv_image = np.array(img) 
+        img = cv2.cvtColor(cv2.imread(open_cv_image), cv2.COLOR_BGR2RGB)
     
         input_ = torch.from_numpy(img).float().div(255.).permute(2,0,1).unsqueeze(0).to(device)