Update app.py

app.py

@@ -5,9 +5,12 @@ from pandasai.llm.starcoder import Starcoder
 import matplotlib.pyplot as plt
 import gradio as gr
 
-def plot_chey(X):
-    fig,ax = X.figure, X.axes
-    return fig
+def plot_chey():
+   a,b = plt.subplots()
+    
+    b.xlabel('two rs')
+    b.title('hada')
+    return a
     
 title='Plot'
 # Define List of Models
@@ -40,7 +43,7 @@ with gr.Blocks(title=title) as demo:
         ''')
     button = gr.Button(value = 'nokku')
     
-    button.click(plot_chey(response), outputs = gr.Plot())
+    button.click(plot_chey(), outputs = gr.Plot())
 demo.launch()