Update app.py

app.py

@@ -5,7 +5,9 @@ from pandasai.llm.starcoder import Starcoder
 import matplotlib.pyplot as plt
 import gradio as gr
 
-
+def plot_chey(X):
+    return X.figure
+    
 title='Plot'
 # Define List of Models
 
@@ -35,7 +37,9 @@ with gr.Blocks(title=title) as demo:
             <h1 style='text-align: center'>Plot</h1>
             </div>
         ''')
-    o1=gr.Plot(response.figure)
+    button = gr.Button(value = 'nokku')
+    
+    button.click(plot_chey(response), outputs = gr.Plot())
 demo.launch()