Tamileditz commited on
Commit
3f026df
·
verified ·
1 Parent(s): ae489fe

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pandas as pd
3
+ from transformers import pipeline
4
+ from pandasai import SmartDataframe
5
+
6
+ llm = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")
7
+
8
+ st.title("Data Analysis with Pandas AI")
9
+
10
+ uploader_file = st.file_uploader("Upload a excel file", type = ["xlsx"] )
11
+
12
+ if uploader_file is not None:
13
+ data = pd.read_excel(uploader_file)
14
+ st.write(data.head(3))
15
+ df = SmartDataframe(data, config={"llm": llm})
16
+ prompt = st.text_area("Enter your Prompt:")
17
+
18
+ if st.button("Generate"):
19
+ if prompt:
20
+ with st.spinner("Generative response ...."):
21
+ st.write(df.chat(prompt))
22
+ else:
23
+ st.warning("Please enter a Prompt")