Update app.py
Browse files
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 = "
|
| 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)
|