arithescientist commited on
Commit
8328a6d
·
verified ·
1 Parent(s): 1c40c30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 GPT-2 model from Hugging Face
14
- llm = pipeline('text-generation', model='gpt2')
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")