jaekookang commited on
Commit
b0e79f5
Β·
1 Parent(s): bcaf154

update plot

Browse files
Files changed (1) hide show
  1. gradio_artist_classifier.py +1 -1
gradio_artist_classifier.py CHANGED
@@ -103,7 +103,7 @@ def predict(input_image):
103
  fig.tight_layout()
104
 
105
  buf = io.BytesIO()
106
- fig.save(buf, bbox_inces='tight', fotmat='jpg')
107
  buf.seek(0)
108
  pil_img = Image.open(buf)
109
  plt.close()
 
103
  fig.tight_layout()
104
 
105
  buf = io.BytesIO()
106
+ fig.savefig(buf, bbox_inces='tight', fotmat='jpg')
107
  buf.seek(0)
108
  pil_img = Image.open(buf)
109
  plt.close()