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

Switch to mistral

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,7 +6,8 @@ import os
6
 
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
 
11
  # In-memory vector index and docs
12
  index = None
 
6
 
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