ning8429 commited on
Commit
973b6f6
·
verified ·
1 Parent(s): 840bb85

Update api_server.py

Browse files
Files changed (1) hide show
  1. api_server.py +1 -1
api_server.py CHANGED
@@ -85,7 +85,7 @@ def predict():
85
 
86
  # 將圖像儲存到一個緩衝區
87
  img_io = io.BytesIO()
88
- image.save(img_io, 'PNG')
89
  img_io.seek(0)
90
 
91
  # 返回圖像
 
85
 
86
  # 將圖像儲存到一個緩衝區
87
  img_io = io.BytesIO()
88
+ image_data.save(img_io, 'PNG')
89
  img_io.seek(0)
90
 
91
  # 返回圖像