SujathaL commited on
Commit
8ced2a4
·
verified ·
1 Parent(s): 2a2dd2f

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 AutoModelForCausalLM, AutoTokenizer
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
 
9
  # Load Mistral-7B Model for Generative Answers
10
- model_name = "mistralai/Mistral-7B-Instruct" # More advanced model
 
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  model = AutoModelForCausalLM.from_pretrained(model_name)
13
 
 
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
 
9
  # Load Mistral-7B Model for Generative Answers
10
+ model_name = "tiiuae/falcon-7b-instruct"
11
+
12
  tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  model = AutoModelForCausalLM.from_pretrained(model_name)
14