Update app.py

app.py

@@ -2,6 +2,7 @@
 import pandas as pd
 from pandasai import PandasAI
 from pandasai.llm.starcoder import Starcoder
+import gradio as gr
 
 # Define List of Models
 
@@ -22,5 +23,6 @@ llm = models[model_to_run](api_token=API_KEY)
 pandas_ai = PandasAI(llm, conversational=False, verbose=True)
 # Enter Prompt related to data or Select from Pre-defined for demo purposes.
 prompt = 'Plot the histogram of countries showing for each the gdp, using different colors for each bar' #@param [ "What is the relation between GDP and Happines Index", "Plot the histogram of countries showing for each the gpd, using different colors for each bar", "GDP of Top 5 Happiest Countries?"]  {allow-input: true}
-pandas_ai.run(df, prompt=prompt,
-                          is_conversational_answer=False)                         
+response = pandas_ai.run(df, prompt=prompt,
+                          is_conversational_answer=False)  
+print(response)