Tamileditz commited on
Commit
e446843
·
verified ·
1 Parent(s): a627724

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import pipeline
5
  from pandasai import SmartDataframe
6
 
7
  #llm = pipeline("text-generation", model="microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True)
8
- llm = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
-
10
 
11
  st.title("Data Analysis with Pandas AI")
12
 
 
5
  from pandasai import SmartDataframe
6
 
7
  #llm = pipeline("text-generation", model="microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True)
8
+ #llm = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
+ llm = pipeline("text-generation", model="microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True)
10
 
11
  st.title("Data Analysis with Pandas AI")
12