mlbench123 commited on
Commit
4580638
·
verified ·
1 Parent(s): ccecede

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -241,7 +241,8 @@ class WarehouseStitcher:
241
  result_pil.save(buf, format='PNG', optimize=True)
242
  buf.seek(0)
243
 
244
- return result_pil, "\n".join(logs), buf
 
245
 
246
 
247
  class PoseGuidedWarehouseStitcher(WarehouseStitcher):
 
241
  result_pil.save(buf, format='PNG', optimize=True)
242
  buf.seek(0)
243
 
244
+ # Return (image preview, logs, downloadable file)
245
+ return result_pil, "\n".join(logs), ("stitched_result.png", buf.getvalue())
246
 
247
 
248
  class PoseGuidedWarehouseStitcher(WarehouseStitcher):