Spaces:
Sleeping
Sleeping
arithescientist
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ import logging
|
|
10 |
if 'history' not in st.session_state:
|
11 |
st.session_state.history = []
|
12 |
|
13 |
-
# Load a pre-trained
|
14 |
-
llm = pipeline('text-generation', model='
|
15 |
|
16 |
# Step 1: Upload CSV data file (or use default)
|
17 |
st.title("Natural Language to SQL Query App with Enhanced Insights")
|
|
|
10 |
if 'history' not in st.session_state:
|
11 |
st.session_state.history = []
|
12 |
|
13 |
+
# Load a smaller and faster pre-trained model (distilgpt2) from Hugging Face
|
14 |
+
llm = pipeline('text-generation', model='distilgpt2') # Using a smaller model for faster inference
|
15 |
|
16 |
# Step 1: Upload CSV data file (or use default)
|
17 |
st.title("Natural Language to SQL Query App with Enhanced Insights")
|