manhkhanhUIT commited on
Commit
601d0a4
1 Parent(s): 7785814

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -22,16 +22,17 @@ def run(image):
22
  cv2.imwrite("Temp/input/input_img.png", image)
23
 
24
  command = ("python run.py --input_folder "
25
- + "Temp/input"
26
- + " --output_folder "
27
- + "Temp"
28
- + " --GPU "
29
- + "-1"
30
- + " --with_scratch")
31
- run_cmd(command)
32
 
33
  result = cv2.imread("Temp/final_output/input_img.png")
34
  shutil.rmtree("Temp")
 
35
  return result
36
 
37
- iface = gr.Interface(fn=run, inputs="image", outputs="image").launch(debug=True, share=True)
 
22
  cv2.imwrite("Temp/input/input_img.png", image)
23
 
24
  command = ("python run.py --input_folder "
25
+ + "Temp/input"
26
+ + " --output_folder "
27
+ + "Temp"
28
+ + " --GPU "
29
+ + "-1"
30
+ + " --with_scratch")
31
+ run_cmd(command)
32
 
33
  result = cv2.imread("Temp/final_output/input_img.png")
34
  shutil.rmtree("Temp")
35
+
36
  return result
37
 
38
+ iface = gr.Interface(fn=run, inputs="image", outputs="image").launch(debug=True)