Spaces:
Running
on
Zero
Running
on
Zero
ZhengPeng7
commited on
Correct the predicted output vars of tab_batch.
Browse files
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 |
-
|
|
|
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/*')][:]
|