beihai commited on
Commit
ff11e14
·
1 Parent(s): 7e260c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,7 +1,8 @@
1
  from xml.dom.pulldom import default_bufsize
2
  import os
3
- #os.system("pip install opencv-python")
4
- #import cv2
 
5
  import numpy as np
6
  from PIL import Image
7
  import gradio as gr
@@ -9,8 +10,7 @@ from func import rebuild_img
9
 
10
 
11
  def inference(img,k):
12
- #input_img = cv2.imread(img, cv2.IMREAD_COLOR)
13
- input_img = Image.open(img)
14
  #k=gr.inputs.Slider(0, 1, 0.1)
15
  u, sigma, v = np.linalg.svd(input_img[:, :, 0])
16
  R = rebuild_img(u, sigma, v, k)
 
1
  from xml.dom.pulldom import default_bufsize
2
  import os
3
+ os.system("pip install --upgrade pip")
4
+ os.system("pip install opencv-python")
5
+ import cv2
6
  import numpy as np
7
  from PIL import Image
8
  import gradio as gr
 
10
 
11
 
12
  def inference(img,k):
13
+ input_img = cv2.imread(img, cv2.IMREAD_COLOR)
 
14
  #k=gr.inputs.Slider(0, 1, 0.1)
15
  u, sigma, v = np.linalg.svd(input_img[:, :, 0])
16
  R = rebuild_img(u, sigma, v, k)