ZhengPeng7 commited on
Commit
abc33ee
·
1 Parent(s): 2032f41

reorder the return values to make the predicted result as the target by the download button in the UI.

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. app_local.py +1 -1
app.py CHANGED
@@ -290,7 +290,7 @@ def predict(images, resolution, weights_file):
290
  zipf.write(file, os.path.basename(file))
291
  return save_paths, zip_file_path
292
  else:
293
- return (image_masked, image_ori)
294
 
295
 
296
  examples = [[_] for _ in glob('examples/*')][:]
 
290
  zipf.write(file, os.path.basename(file))
291
  return save_paths, zip_file_path
292
  else:
293
+ return (image_ori, image_masked)
294
 
295
 
296
  examples = [[_] for _ in glob('examples/*')][:]
app_local.py CHANGED
@@ -285,7 +285,7 @@ def predict(images, resolution, weights_file):
285
  zipf.write(file, os.path.basename(file))
286
  return save_paths, zip_file_path
287
  else:
288
- return (image_masked, image_ori)
289
 
290
 
291
  examples = [[_] for _ in glob('examples/*')][:]
 
285
  zipf.write(file, os.path.basename(file))
286
  return save_paths, zip_file_path
287
  else:
288
+ return (image_ori, image_masked)
289
 
290
 
291
  examples = [[_] for _ in glob('examples/*')][:]