0x90e commited on
Commit
58ba404
·
1 Parent(s): 3a83ad9
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -29,8 +29,8 @@ def inference(img):
29
  run_cmd("python test.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
30
  return os.path.join(OUTPUT_DIR, "1_out.jpg")
31
 
32
- input_image = gr.inputs.Image(type="pil", label="Input")
33
- output_image = gr.outputs.Image(type="file", label="Output")
34
 
35
  demo = gr.Interface(
36
  inference,
@@ -38,6 +38,4 @@ demo = gr.Interface(
38
  outputs=[output_image]
39
  )
40
 
41
- demo.output(save_image=True, save_image_format='jpg')
42
-
43
  demo.launch(debug=is_colab, share=is_colab)
 
29
  run_cmd("python test.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
30
  return os.path.join(OUTPUT_DIR, "1_out.jpg")
31
 
32
+ input_image = gr.Image(type="pil", label="Input")
33
+ output_image = gr.Image(type="filepath")
34
 
35
  demo = gr.Interface(
36
  inference,
 
38
  outputs=[output_image]
39
  )
40
 
 
 
41
  demo.launch(debug=is_colab, share=is_colab)