Spaces:
Runtime error
Runtime error
Xu Ma
commited on
Commit
•
461b439
1
Parent(s):
a85a5c8
update
Browse files
app.py
CHANGED
@@ -297,7 +297,7 @@ def main(args):
|
|
297 |
experiment_id, # path adding scheduler
|
298 |
]
|
299 |
# outputs
|
300 |
-
outputs = gr.outputs.Image(type="
|
301 |
outputs02 = gr.outputs.JSON(label="检测信息")
|
302 |
|
303 |
# title
|
|
|
297 |
experiment_id, # path adding scheduler
|
298 |
]
|
299 |
# outputs
|
300 |
+
outputs = gr.outputs.Image(type="pil", label="检测图片")
|
301 |
outputs02 = gr.outputs.JSON(label="检测信息")
|
302 |
|
303 |
# title
|
main.py
CHANGED
@@ -522,7 +522,7 @@ def main_func(target, experiment, cfg_arg):
|
|
522 |
### added for app
|
523 |
if t%10==0:
|
524 |
print(f"debug: {t}, {filename}")
|
525 |
-
return
|
526 |
|
527 |
x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW
|
528 |
|
|
|
522 |
### added for app
|
523 |
if t%10==0:
|
524 |
print(f"debug: {t}, {filename}")
|
525 |
+
return img, None
|
526 |
|
527 |
x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW
|
528 |
|