Hugging Face's logo

Spaces:
saadkhi
/
SQL_chatbot_API
Running

App Files Community
saadkhi commited on
Commit
59d4862
·
verified ·
1 Parent(s): be03b42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
7
 
8
  torch.set_num_threads(1)
9
 
10
- BASE_MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 
11
 
12
  print("Loading model...")
13
 
 
7
 
8
  torch.set_num_threads(1)
9
 
10
+ # BASE_MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
11
+ BASE_MODEL = "tiiuae/falcon-rw-1b"
12
 
13
  print("Loading model...")
14