w11wo commited on
Commit
9982fb2
1 Parent(s): 6e92463

initial commit

Browse files
app.py CHANGED
@@ -1,10 +1,6 @@
1
  import numpy as np
2
  import cv2
3
  import onnxruntime
4
- from glob import glob
5
- import os
6
- import gradio as gr
7
-
8
  import gradio as gr
9
 
10
 
@@ -41,7 +37,7 @@ def convert_pil_to_cv2(image):
41
 
42
 
43
  def main(image):
44
- model_path = "./model.ort"
45
  img = convert_pil_to_cv2(image)
46
  if img.ndim == 2:
47
  img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
 
1
  import numpy as np
2
  import cv2
3
  import onnxruntime
 
 
 
 
4
  import gradio as gr
5
 
6
 
 
37
 
38
 
39
  def main(image):
40
+ model_path = "models/model.ort"
41
  img = convert_pil_to_cv2(image)
42
  if img.ndim == 2:
43
  img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
example1_x2.jpg → examples/example1_x2.jpg RENAMED
File without changes
model.ort → models/model.ort RENAMED
File without changes
requirements.txt CHANGED
@@ -1,3 +1,2 @@
1
  numpy
2
- onnxruntime
3
- opencv-python
 
1
  numpy
2
+ onnxruntime