ZhengPeng7 commited on
Commit
8a9ec25
1 Parent(s): 5d10050

Correct the predicted output vars of tab_batch.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -134,8 +134,9 @@ def predict(images, resolution, weights_file):
134
  with zipfile.ZipFile(zip_file_path, 'w') as zipf:
135
  for file in save_paths:
136
  zipf.write(file, os.path.basename(file))
137
-
138
- return image, image_masked
 
139
 
140
 
141
  examples = [[_] for _ in glob('examples/*')][:]
 
134
  with zipfile.ZipFile(zip_file_path, 'w') as zipf:
135
  for file in save_paths:
136
  zipf.write(file, os.path.basename(file))
137
+ return save_paths, zip_file_path
138
+ else:
139
+ return image, image_masked
140
 
141
 
142
  examples = [[_] for _ in glob('examples/*')][:]