Spaces:
Running
on
Zero
Running
on
Zero
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/*')][:]
|