Veda0718 commited on
Commit
0abfdab
·
verified ·
1 Parent(s): 968fa1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -61,6 +61,7 @@ class CSVChatbot:
61
  result = self.agent.run(CSV_PROMPT_PREFIX + message + CSV_PROMPT_SUFFIX)
62
  fig = plt.gcf()
63
  if fig.get_axes():
 
64
  buf = io.BytesIO()
65
  fig.savefig(buf, format='png')
66
  buf.seek(0)
 
61
  result = self.agent.run(CSV_PROMPT_PREFIX + message + CSV_PROMPT_SUFFIX)
62
  fig = plt.gcf()
63
  if fig.get_axes():
64
+ plt.clf()
65
  buf = io.BytesIO()
66
  fig.savefig(buf, format='png')
67
  buf.seek(0)