sourize commited on
Commit
939ae3f
·
verified ·
1 Parent(s): 7b7d4e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ def load_resources():
11
  embedder = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
12
  chat_gen = pipeline(
13
  'text2text-generation',
14
- model='microsoft/phi-2',
15
- tokenizer='microsoft/phi-2',
16
  device=-1,
17
  # enforce deterministic decoding and low temperature to reduce hallucinations
18
  do_sample=False,
 
11
  embedder = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
12
  chat_gen = pipeline(
13
  'text2text-generation',
14
+ model='google/flan-t5-base',
15
+ tokenizer='google/flan-t5-base',
16
  device=-1,
17
  # enforce deterministic decoding and low temperature to reduce hallucinations
18
  do_sample=False,