noamholz commited on
Commit
36f4824
·
verified ·
1 Parent(s): 5ee3cb8

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -0
run.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import numpy as np
3
  from time import sleep
4
  import torch
 
5
  from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
6
  # from torchvision import transforms
7
 
 
2
  import numpy as np
3
  from time import sleep
4
  import torch
5
+ import cv2
6
  from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
7
  # from torchvision import transforms
8