nathanjc commited on
Commit
2f472db
·
1 Parent(s): 74e1781

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -202,7 +202,7 @@ import numpy as np
202
  # self.exist_ok=exist_ok
203
  # self.no_trace=no_trace
204
 
205
- def get_output(image):
206
  # image.save(f"{BASE_DIR}/input/image.jpg")
207
  # source = f"{BASE_DIR}/input"
208
  # opt = options(weights='logo_detection.pt',source=source)
@@ -212,7 +212,7 @@ def get_output(image):
212
  # if os.path.exists(output_path):
213
  # return np.array(Image.open(output_path))
214
  # else:
215
- return image
216
 
217
 
218
  demo = gr.Interface(fn=get_output, inputs="image", outputs="image")
 
202
  # self.exist_ok=exist_ok
203
  # self.no_trace=no_trace
204
 
205
+ def get_output(input_image):
206
  # image.save(f"{BASE_DIR}/input/image.jpg")
207
  # source = f"{BASE_DIR}/input"
208
  # opt = options(weights='logo_detection.pt',source=source)
 
212
  # if os.path.exists(output_path):
213
  # return np.array(Image.open(output_path))
214
  # else:
215
+ return input_image
216
 
217
 
218
  demo = gr.Interface(fn=get_output, inputs="image", outputs="image")