Xu Ma commited on
Commit
461b439
1 Parent(s): a85a5c8
Files changed (2) hide show
  1. app.py +1 -1
  2. main.py +1 -1
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="file", label="检测图片")
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 filename, None
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