Abhishek Gola commited on
Commit
30d469c
·
1 Parent(s): f70c427

Updated app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -10,7 +10,9 @@ model_path = hf_hub_download(repo_id="opencv/edge_detection_dexined", filename="
10
  model = Dexined(modelPath=model_path)
11
 
12
  def detect_edges(input_image):
 
13
  result = model.infer(input_image)
 
14
  return result
15
 
16
  # Gradio UI
 
10
  model = Dexined(modelPath=model_path)
11
 
12
  def detect_edges(input_image):
13
+ input_image = cv.cvtColor(input_image, cv.COLOR_RGB2BGR)
14
  result = model.infer(input_image)
15
+ result = cv.cvtColor(result, cv.COLOR_BGR2RGB)
16
  return result
17
 
18
  # Gradio UI