StevenMSAI commited on
Commit
d8c5a1c
·
verified ·
1 Parent(s): 6092044

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from transformers import pipeline
13
  # cores and 16gb ram only. So we need to keep these lightweight + cpu-only
14
 
15
  EMBED_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2" # small & fast on CPU
16
- GEN_MODEL_NAME = "google/flan-t5-small" # text2text model that runs on CPU
17
 
18
  embedder = SentenceTransformer(EMBED_MODEL_NAME)
19
  generator = pipeline("text2text-generation", model=GEN_MODEL_NAME)
 
13
  # cores and 16gb ram only. So we need to keep these lightweight + cpu-only
14
 
15
  EMBED_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2" # small & fast on CPU
16
+ GEN_MODEL_NAME = "MBZUAI/LaMini-Flan-T5-248M" # text2text model that runs on CPU
17
 
18
  embedder = SentenceTransformer(EMBED_MODEL_NAME)
19
  generator = pipeline("text2text-generation", model=GEN_MODEL_NAME)