Update app.py

app.py

@@ -5,8 +5,7 @@ from pandasai.llm.starcoder import Starcoder
 import matplotlib.pyplot as plt
 import gradio as gr
 
-def make_plot(x):
-    return x.figure
+
 title='Plot'
 # Define List of Models
 
@@ -36,6 +35,7 @@ with gr.Blocks(title=title) as demo:
             <h1 style='text-align: center'>Plot</h1>
             </div>
         ''')
-    o1=gr.Plot()
-    btn = gr.Button(value = 'Submit')
-    btn.click(make_plot,inputs=[btn],outputs=[o1])
+    o1=gr.Plot(response.figure)
+demo.launch()
+    
+