Joe Hare commited on
Commit
a3272dc
·
1 Parent(s): 0e7f45b

Switch to falcon

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import os
7
  # Load models
8
  embedder = SentenceTransformer("all-MiniLM-L6-v2")
9
  #qa_pipeline = pipeline("text-generation", model="gpt2") # Replace with a better QA model if desired
10
- qa_pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
11
 
12
  # In-memory vector index and docs
13
  index = None
 
7
  # Load models
8
  embedder = SentenceTransformer("all-MiniLM-L6-v2")
9
  #qa_pipeline = pipeline("text-generation", model="gpt2") # Replace with a better QA model if desired
10
+ qa_pipeline = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
11
 
12
  # In-memory vector index and docs
13
  index = None