Update app.py

app.py

@@ -3,7 +3,9 @@ import pandas as pd
 from transformers import pipeline
 from pandasai import SmartDataframe
 
-llm =  pipeline("text-generation", model="microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True)
+#llm =  pipeline("text-generation", model="microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True)
+llm = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
+
 
 st.title("Data Analysis with Pandas AI")